ubunt16.04 安装3090显卡驱动 cuda cudnn pytorch

安装驱动

需要的安装包

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