【在anaconda的虚拟环境中一键安装tensorflow+cuda+cudnn】

一键安装指令

conda install tensorflow-gpu=1.8 cudatoolkit=9.0 cudnn=7.0

升级指令

# 升级tensorflow
pip install --upgrade tensorflow-gpu==1.15

# 升级cuda。先卸载cuda,再安装新的cuda
conda uninstall cudatoolkit
conda install cudatoolkit=10.0

# 升级cudnn。先卸载cuda,再安装新的cudnn
conda uninstall cudnn
conda install cudnn=7.4
# 结果发现找不到7.4版本,所以用另外的网站来安装
conda install cudnn=7.4.1 -c https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installwindows

## 重命名anaconda环境名
#例如:Tensorflow1.8环境重命名为tensorflow1.15
# 1、进入旧环境
activate Tensorflow1.8
# 2、克隆旧环境,新环境名为tensorflow1.15
conda create -n tensorflow1.15 --clone Tensorflow1.8
# 3、退出旧环境
conda deactivate
# 4、删除旧环境
conda remove -n Tensorflow1.8 --all
# 5、检查是否重命名成功
conda info --envs

# 记得更改新环境的三个环境变量路径名称,否则pycharm找不到新环境的库。

# 检查tf的gpu是否可用
import  tensorflow as tf
print(tf.test.is_gpu_available())

版本对应

【在anaconda的虚拟环境中一键安装tensorflow+cuda+cudnn】_第1张图片

参考链接
anaconda修改环境名称

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