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

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

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

1.安装CUDA

桌面右键-NVIDIA控制面板-帮助-系统信息-组件，可以查看自己的显卡支持的最高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