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!

深度学习环境配置2——windows下的torch=1.2.0环境配置_Bubbliiiing的博客-CSDN博客_torch环境配置

猜你喜欢

转载自blog.csdn.net/m0_63172128/article/details/129198848