【win10配置pytorch深度学习环境全记录】

2021-01-23 21:01:14 阅读：327 来源： 互联网

标签：全记录 pytorch conda CUDA anaconda win10 镜像安装

寒假在家，只好在自己的笔记本上配置pytorch的环境。

win10配置pytorch深度学习环境全记录

1.安装CUDA

桌面右键-NVIDIA控制面板-帮助-系统信息-组件，可以查看自己的显卡支持的最高CUDA版本
查看CUDA版本
可以看到最高支持到CUDA11.0，于是决定安装CUDA11.0
CUDA11.0官网链接
如下图勾选，下载后即可安装
在这里插入图片描述

安装过程详见安装CUDA11.0的参考文章
文章中说需要先安装VS2019，并装在C盘默认目录中。由于C盘空间有限，选择安装在了D盘中，也没有出现问题
安装后，按照参考文章中的方法配置路径并验证是否安装成功即可

2.配置pytorch

anaconda和python的安装不再赘述
pytorch的安装直接去复制命令安装时最方便的，但是如果使用官方源会非常慢，建议使用国内镜像源
在Anaconda Prompt中输入以下指令可以添加镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/

到pytorch官网复制安装pytorch的指令

安装指令

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

在这里遇到一个问题，conda在安装时找到的pytorch还是官网的路径，而不是镜像源的，速度很慢
只好去清华镜像源下载了pytorch，然后离线安装
清华镜像源

先离线安装好pytorch，再执行从大安装指令即可
至此，安装完成
在python环境中执行

import torch
torch.cuda.is_available()

如果结果为True，说明安装成功

3.总结

希望能帮助到由同样需求的人
同时作为笔记

参考：
CUDA安装参考1
CUDA安装参考2
CUDA安装参考3
pytorch官网
 清华pytorch源
 pytorch安装

标签：全记录,pytorch,conda,CUDA,anaconda,win10,镜像,安装
来源： https://blog.csdn.net/East_Plain/article/details/113061373

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

【win10配置pytorch深度学习环境全记录】