[人工智能]pytorch+gpu安装两步完成

前话

网上一堆教程，安装这个安装那个，那么麻烦，今天我发现完成这个安装实在是太简单了。

1.查看CUDA版本

这个有好几种方法，我只提供一种：鼠标右键或者通过右下角箭头，进入到NIVIDIA控制面板，

然后点击系统信息

点击‘组件’，即可看到自己的cuda版本。

2.下载pytorch

进入pytorch下载官网，查看自己的版本
看箭头标的位置，说明这个命令会自动帮忙安装cuda ，就不需要管那么多了 ，然后输入这里的命令就可以了。

3.完成后可以测试一下

pytorch

import torch
x = torch.empty(3,5)
print(x)

CUDA

print(torch.cuda.is_available())  

如果结果是True即可。

CUDNN

from torch.backends import  cudnn 
print(cudnn.is_available())

如果结果是True即可。

下面就是我的结果了，很nice