安装tensorflow-GPU

前言

不想用tensorflow,但是有的代码,没有torch版本的,只能捡起来tensorflow。

参考

https://blog.csdn.net/m0_68233404/article/details/128525860

电脑配置

项目 Value
OS Windows
GPU RTX3060
CUDA 11.2
python 3.8
Anaconda 参考

Anconda构建python=3.8虚拟环境

# 创建
conda create --prefix=tensorflowGPUEnv python=3.8
# 激活
conda activate     tensorflowGPUEnv的绝对路径

安装requirements

在requirements中的tensorflow版本与自己电脑cuda版本匹配的情况下,可以使用

pip install -r   requirements的绝对路径 -i https://pypi.tuna.tsinghua.edu.cn/simple

手动安装tensorflow

tensorflow版本对应

安装tensorflow-GPU_第1张图片

待更新, 因为我直接安装requirement的了

你可能感兴趣的:(tensorflow,python,深度学习)