安装驱动

需要的安装包

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