一、说明
pytorch1.2.0+python3.6+CUDA=10.0+cudnn=7.4.1.5
二、步骤
在conda中创建一个新的虚拟环境
查看一下自己的所有环境
激活虚拟环境
conda activate torch1.2.0
关于cuda和cudnn
1、查看自己电脑系统是10.2版本
http://链接:https://pan.baidu.com/s/1v5cN6TbgZg2Hkarc60_KrQ?pwd=oh9v 提取码:oh9vhttp://xn--https-bl8js66z7n7i//pan.baidu.com/s/1v5cN6TbgZg2Hkarc60_KrQ?pwd=oh9v%20%20%E6%8F%90%E5%8F%96%E7%A0%81%EF%BC%9Aoh9v
想直接下载安装对应10.0版本,结果在自定义路径不能和已有的(上图)放一个路径,可能是系统一个路径只能放一个的原因,于是想在虚拟环境中安装,查资料说虚拟环境和系统的cuda不会有影响?
安CUDA:
(1)搜索库名:cuda,命令如下:
anaconda search -t conda cudatoolkit
安cudnn:
anaconda show conda-forge/cudnn
没有+7.4.1.5怎么办?
https://blog.csdn.net/qq_43705697/article/details/121618276
conda search cudnn --info
我最后安装了推荐的cudnn=7.6.5.32
conda install --channel https://conda.anaconda.org/conda-forge cudnn=7.6.5.32
测试安装是否成功
在虚拟环境中安装完cuda和cudnn想要测试是否安装成功,不能使用nvcc -V命令测试,需要在虚拟环境中安装pytorch包进行测试
安装pytorch
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
接下来安装一些包:保存下面到一个txt文本文件,然后找到路径,如下方法一键安装:
scipy==1.2.1
numpy==1.17.0
matplotlib==3.1.2
opencv_python==4.1.2.30
torch==1.2.0
torchvision==0.4.0
tqdm==4.60.0
Pillow==8.2.0
h5py==2.10.0
pip install -r C:\Users\hp\Desktop\1.txt