anaconda虚拟环境中安装特定版本cuda+tensorflow,之前自己安装过,但是没有总结,后面又需要这样安装,结果就忘得干干净净,这次决定总结下操作步骤:
1.安装清华源(如果安装过,请掠过,如果第二步骤出现问题,那么需要清除源,命令为conda config --remove-key channels,并且执行此步骤),使用国内源更快:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
2.安装CUDA和cudnn,命令如下:
#注意:这里的cuda版本和你要安装的tensorflow版本要适配 , 这里给出一个链接提供参考: https://blog.csdn.net/weixin_42206075/article/details/122998744
conda install cudatoolkit=10.0 # 指定你自己的版本
conda install cudnn=7.6 # 指定你自己的版本
3.使用pip安装自己需要的tensorflow或则tensorflow-gpu版本,也可以使用添加清华源的安装,如下:
# pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple 模块名称
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0
4.安装其他模块,清华源安装参照第3步.
5.如果查看是否安装成功,执行下面代码:
# Pytorch
import torch
print(torch.cuda.is_available())
# Tensorflow
import tensorflow as tf
print(tf.test.is_gpu_available())
注意：欢迎转载，转载时请注明来源
