win10安装pytorch并解决torch.cuda.is_available()为False的问题

安装pytorch

环境：win10、pytorch1.6.0、cuda10.1、cudnn7.6.5

安装步骤（打开 anaconda powershell prompt）：

cuda与cndnn的安装网络上有很多，在此不做赘述

1. 添加清华镜像源

依次输入以下命令：
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
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 --show channels

2. 新建虚拟环境

新建：
conda create --name pytorch_env python=3.6
激活：
conda activate pytorch_env

3. 安装pytorch (耐心等待)

conda install pytorch torchvision cudatoolkit=10.1

4. 检查是否安装成功(本人在此处返回了False)

输入python进入命令行
>>> import torch
>>> torch.cuda.is_available()
若返回True则成功，若返回False则失败

解决torch.cuda.is_available()为False的问题

问题：电脑安装的显卡驱动低于CUDA版本所需

• 查看电脑的显卡驱动版本

  1. win+r，输入“cmd”打开“命令行提示符”窗口
  2. 输入：nvidia-smi查看系统的显卡驱动信息
  3. 发现 CUDA Version: 10.0 低于所安装的cuda10.1版本
     

• 升级系统的显卡驱动

  1. 下载驱动精灵
  2. 打开驱动管理，找到
  3. 点击升级，下载完成之后，一步步执行安装即可
  4. 重新在命令行输入nvidia-smi，发现已更新为CUDA Version: 10.2
     

• 再次检查是否安装成功，发现torch.cuda.is_available()返回为True

注：除了升级系统驱动，也可以选择降低cuda的安装版本