Win10 下安装cuda 10.2及pytorch 1.7

记录一下win10 下安装cuda 10.2及pytorch 1.7

1. 先去英伟达官网下载cuda 10.2以及cudann, 其中cudacnn的版本根据cuda 10.2来对应选择。这里官网下载速度过慢，建议使用迅雷下载，右键复制出下载链接然后迅雷下载即可。
   

   • cuda 10.2 网址：https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

   • cudann网址：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse803-102

   • 检查是否安装成功的命令，网上资料也很详细：

     nvcc --version
     

它们具体的安装细节请参考其他博客。如果有些驱动安装失败，那就换cuda版本吧，显卡上给的都是最高支持的版本，我的虽然支持cuda11， 但是有一个驱动安装一直失败，我就装cuda10.2了。

2. 安装Anaonda，这里很简单，正常人不会安装出错；这里推荐北外的镜像网址：https://mirrors.bfsu.edu.cn/anaconda/archive/?C=M&O=D

3. 最后安装pytoch

   • 第一步是修改镜像：Windows 用户无法直接创建名为 .condarc 的文件，可在anaconda promt先执行 conda config --set show_channel_urls yes 生成该文件之后再修改。

   conda config --set show_channel_urls yes
   

   • 然后去路径：C:\Users\Lyons找文件.condarc；说明一下：Lyons是我的电脑用户名，且我的anaconda装在C:\Users\Lyons\anaconda3；用笔记本打开文件.condarc，我的修改为：

   show_channel_urls: true
   channels:
     - https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
     - https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
     - https://mirrors.bfsu.edu.cn//anaconda/cloud/pytorch/
     - defaults
   

   • 这个pytorch要与cuda版本一一对应，建议去pytorch官网看安装命令：https://pytorch.org/get-started/locally/

     由于要选择镜像来进行下载，故把命令后面的 -c pytorch去掉

   conda install pytorch torchvision torchaudio cudatoolkit=10.2
   

   北外的下载速度还是很给力的，最后检查pytorch能否使用GPU：

   import torch 
   torch.cuda.is_available()