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