anaconda环境下配置pytorch要点概述

一、下载anaconda

直接在官网上下载电脑匹配版本anaconda，值得注意的是，需要手动将anaconda下载路径中的Scrips添加到电脑环境变量中，不然会出现无法识别conda命令的情况。

二、下载cuda和cudnn

需要这两个调用gpu。

先Win+R输入cmd进入命令行，输入NVIDIA-smi查看电脑gpu的版本和driver版本。再根据下链接中找到对应的cuda和cudnn下载。

(59条消息) pytorch libtorch 不同版本对应的cuda_今年不吃饭...的博客-CSDN博客s​​​​​​​shttps://blog.csdn.net/agq358/article/details/121074551?ops_request_misc=&request_id=&biz_id=102&utm_term=cuda10.2%E5%AF%B9%E5%BA%94%E7%9A%84pytorch&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-6-121074551.nonecase&spm=1018.2226.3001.4187三、安装python

不同版本cuda支持的python不同，根据上链接可以找到对应版本Python,在官网（python.org）中找到并下载即可

引自：

(59条消息) Python安装教程-史上最全_壬杰的博客-CSDN博客https://blog.csdn.net/weixin_49237144/article/details/122915089?ops_request_misc=&request_id=&biz_id=102&utm_term=python%E5%AE%89%E8%A3%85&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-0-122915089.nonecase&spm=1018.2226.3001.4187 

四、设置清华镜像源以便后续安装pytorch

(59条消息) CondaHTTPError: HTTP 000 CONNECTION FAILED for url ＜https://repo.anaconda.co_王胖胖要加油鸭的博客-CSDN博客https://blog.csdn.net/qq_35443700/article/details/107951536?spm=1001.2014.3001.5502

五、配置环境

Win+R cmd 进入命令行模式

conda create -name pytorch python=3.8（你的python版本号）

y  #确认创建环境

conda info --envs            #可以看到anaconda中的环境，base和pytorch

在pytorch官网（pytorch.org）中找到你所安装cuda对应版本的pytorch

ctrl+f搜索CUDA 10.2(你安装的版本号)，可以找到如下内容。

 复制conda install pytorch==1.12.1 torchvision==……不要最后的-c pytorch，到命令行中，安装pytorch。

 

conda activate pytorch     #激活并进入pytorch环境

python   #进入Python

import torch #检验pytorch是否安装成功 不报错则可用

torch.cuda.is_available() #验证gpu是否可用 如果返回Ture则可用

六、总结

版本！版本！版本！

cuda和cudnn版本匹配，并和电脑版本匹配。

python和安装的cuda版本匹配。

pytorch和安装的cuda版本匹配。

版本匹配和镜像源问题处理好剩下的水到渠成。