一、下载anaconda
直接在官网上下载电脑匹配版本anaconda,值得注意的是,需要手动将anaconda下载路径中的Scrips添加到电脑环境变量中,不然会出现无法识别conda命令的情况。
二、下载cuda和cudnn
需要这两个调用gpu。
先Win+R输入cmd进入命令行,输入NVIDIA-smi查看电脑gpu的版本和driver版本。再根据下链接中找到对应的cuda和cudnn下载。
(59条消息) pytorch libtorch 不同版本对应的cuda_今年不吃饭...的博客-CSDN博客sshttps://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版本匹配。
版本匹配和镜像源问题处理好剩下的水到渠成。