Win10 pytorch GPU版本安装

1.  查看显卡型号       nvidia-smi       --->

2.  安装Anacodna  这里选择5.2.0版本   老版本在这里

查看Anaconda 是否安装成功

出现（base）就安装成功了

接下来市重头戏----》pytorch 的安装，一般官网为默认最新版，但是会发现默认选项CUDA只有10.2 和11.1 而我的是10.1 我先自动安装下看看能不能用。如需查看CUDA对应的torch版本在默认的下方有之前的版本链接：https://pytorch.org/get-started/previous-versions/。另外也有网上大神的指定cuda和pytorch版本安装如下：https://blog.youkuaiyun.com/kellyroslyn/article/details/109668001

 

安装之后查看安装情况!!   意思就是torch 和torchvision 安好的但是CUDA 不能使用！！重装吧。。。

 我们根据CUDA10.1 选择一个版本如 pytorch==1.7.1

安装后进行查询应该是成功了hahahah