安装前需要安装显卡驱动 +cuda+cudnn(当然python也的有)如果这都安装好了就很快了,我这装的是pip版本
首先进入到官网
https://pytorch.org/get-started/locally/
声明以下 我系统是ubunu20.04的,然后cuda是12.0的版本 cudnn是8.0的,然后下载这个pytorch的时候找半天找不到对应cuda12.0的只有最高11.8我看了看发行版也没有对应12.0的这时候心想MMP完蛋了,难不成再来一次??
对于我第一次装这个也太难了,然后拼命百度,发现有博主说这个pytorch说的cuda版本只要大于等于就行了 ,一下高兴了,接着进行科学上网把他下载下来了。顺利安装
检测一下是不是成功了
首先:
python3
import torch
print(torch.cuda.is_available())
出现true就OK!!!!!!
看以显卡刷新
watch -n 1 nvidia-smi
后台挂起 显卡资源一秒一刷新能看到
B站 视频
https://www.bilibili.com/video/BV13Y411T71Y/?spm_id_from=333.788&vd_source=cb680604d81057c520691bbb28e0b570