pytorch-gpu版本安装

1.gpu驱动安装：

在nvidia官网下载符合自己gpu型号的驱动，chmod+x增加可执行权限，然后用./xxx.run安装 过程中提示gcc版本4.8.5不符合，需要安装7.4.0，教程：https://blog.csdn.net/xs1102/article/details/89175293 安装之后还是显示gcc compil是4.8，使用以下教程：https://blog.csdn.net/hizengbiao/article/details/103698560 https://www.jianshu.com/p/f7c789e2ad97 之后按照界面提示继续安装即可，教程可参考：https://blog.csdn.net/ksws0292756/article/details/79160742

2.cuda+cudnn安装

直接在nvidia上面下载安装 验证版本号的指令是nvcc --version 参考教程：https://blog.csdn.net/qq_33200967/article/details/80689543 https://blog.csdn.net/weixin_38632246/article/details/96137803

3.pytorch安装：

直接参考pytorch界面中的指令进行安装即可。

4.检查pytorch安装版本号

import torch print(torch.__version__)

5.测试pytorch的GPU版本是否安装成功

import torch print(torch.cuda.is_available())

6.查看显卡驱动:

nvidia-smi

7.怎样查看本机显卡型号:

lspci | grep -i nvidia