pytorch GPU版本安装从1.0.1到1.1.0之后的版本

• 安装GPU的版本的pytorch最好使用Anaconda
• 需注意，1.1.0之后的版本停止了cuda8.0的支持，所以不要踩雷了，你为啥装不了1.1之后的版本原因在这了。
• 你想装1.1之后的版本先要去更新cuda的版本，这里面可能还需要你更新驱动程序。看看你的驱动版本能不能安装对应的cuda，cmd命令如下：nvidia-smi，红色的圈就是你的驱动版本号，

• 我的版本号为442.59，参照下表，我可以安装到最新版本的CUDA。理论上向下支持的，但是实测我只能安装最新版本的CUDA10.2，我也不知为何，安装其他版本的就失败了。

• 下面你还要安装cudnn，很简单，下载地址出，（麻烦的是你需要注册英伟达账号），对应到你的cuda版本下载。

                          https://developer.nvidia.com/rdp/cudnn-download

• 下载后解压，里面三个文件夹，拷贝到你刚刚安装的cuda目录下，覆盖

cuda目录在哪，一般在c盘，我的在这

 

• 冗长的铺垫做完了，进入安装pytorch环节
• 使用清华源快一点，不然速度就是龟速，cmd输入下面命令

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

安装命令为：

conda install pytorch torchvision cudatoolkit=10.0

• 这里我安装的cudatoolkit为10.0版本的，具体看你cuda的最高版本，往下是支持的，但你选择了8.0的话那你只能安装的最高版本pytorch就为1.0.1了。
• 等待就成功了
• 
• 我爬过坑坑洼洼的沟，只为给你铺平前方的路。我是灰哥，记录无价，请给个赞赞👍，谢谢！