安装GPU版本的pytorch

Pytorch目前主要在学术研究方向领域处于领先地位 。PyTorch可以使用强大的GPU加速的Tensor计算（比如：Numpy的使用）以及可以构建带有autograd的深度神经网络。

同时，PyTorch 的代码很简洁、易于使用、支持计算过程中的动态图而且内存使用很高效。

当在用GPU跑一个pytorch的项目时，检查发现：

#cuda是否可用；
torch.cuda.is_available()

上述代码检查cuda是否可用，运行结果如下：

复制代码

out：False

然后我们打印一下当前版本的torch，到底下载的是CPU版本的还是GPU版本的

import torch
print(torch.version)

运行结果:

复制代码

out: 1.7.1+cpu

+cpu表示当前torch的版本是CPU的。另外，也可以在你的anaconda目录下查看torch的版本，例如我的在

env目录下是你的环境，这里我的环境是叫d2l，如果不是在其他环境下那么就直接在miniconda/lib/site-packages中找到

torch-1.1.1+cpu.dist.info + c p u 表示 C P U 版本

torch-1.7.1+cu101.dist-info + cu101 表示 CUDA 10.1

如果发现是cpu版本的torch，那么需要把当前的torch及torchvision删掉或者卸载

pip uninstall torch torchvision
查看CUDA Version

NVIDIA-smi

可以看到我的当前CUDA版本是 12.8

进入pytorch的官网下载对应cuda版本的torch和torchvision

编辑PyTorchpytorch.org/

下面是打开后的页面（注意可能很卡，建议挂梯子），选择对应版本的系统，建议使用conda下载（比起pip，conda可以帮你把相关的其他安装包一并装上，还会考虑兼容性问题 ），语言"python"，computer platform选择对应版本的cuda，如果没有找到，点击左下角的"previous versions of pytorch"，里面有之前的各个版本

复制命令到anaconda 相应环境下运行：

验证torch是否安装成功

import torch
print(torch.version)

显示出对应版本的torch就说明安装成功。

接着，验证cuda到底能不能用

#cuda是否可用；
torch.cuda.is_available()

运行结果

复制代码

out: True

问题解决！此时就可以用GPU来跑程序啦！