pytorch无法使用显卡运行程序的解决办法

测试Cuda版本

win+R打开运行窗口，输入cmd打开命令窗口

输入命令

>nvcc -V

显示如下结果，证明cuda已经安装成功

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243

测试Anaconda版本并创建环境

在cmd窗口输入指令

>conda --version

返回版本号，证明安装成功

打开Anaconda窗口，选择Environment，点击creat创建自己需要的python版本，我创建的是3.7版本，环境名称自己取一个好记的

安装pytorch-gpu

前往Pytorch官网，按照自己需要的配置进行选择，第一项可以默认选择稳定版，选择完成后复制生成的下载命令。
在cmd窗口激活自己的环境，放入刚才粘贴的代码，等待安装

>activate （你的环境名称）
>conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

安装完成后测试显卡能否使用

>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>import torch
>>>torch.cuda.is_available()
True

显示True证明成功