安装驱动
需要的安装包
30系列显卡是新一代架构,新驱动不支持cuda9以及cuda10,所以必须安装cuda11、而pytorch现在稳定版为1.6,最高仅支持到cud10.2。所以唯一的办法就是使用上处于beta测试的1.7或1.8。这也是为啥一开始就强调本文的写作时间,因为未来一定会有稳定版发布,请使用稳定版。torch_geometric 计算包官方发布的现成轮子仅支持到pytorch-1.6和cuda10.2。
所以核心问题就是cuda11 pytorch 以及torch_geometric之间的匹配。
环境介绍
ubuntu 20.04
Nvidia 驱动版本 455.23
cuda 11.1
cudnn-11.1-linux-x64-v8.0.4.30
pytorch 1.8.0a0+fe4f90c
torchvision 0.8
安装cuda
sudo cuda_11.0.176_384.81_linux.run
安装cudnn
二,deb安装
a,安装runtime library
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
b,安装developer library
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
c,安装code samples and the cuDNN Library User Guide
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
3,验证
a,拷贝示例代码到任意可写路径
$cp -r /usr/src/cudnn_samples_v7/ .
b,进入cudnn_samples_v7相应目录
$ cd cudnn_samples_v7/mnistCUDNN
c,编译
$make clean &&& make
d.运行
$ ./mnistCUDNN
安装pytorch
conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly
测试
参考
/weixin_43402775/article/details/107610267
/qq_15602569/article/details/109232265