安装驱动
需要的安装包
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
测试
参考
https://blog.csdn.net/weixin_43402775/article/details/107610267
https://blog.csdn.net/qq_15602569/article/details/109232265
