pytorch1.2.0+python3.6

一、说明

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

测试：

 ok！