安装驱动

需要的安装包

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

ubunt16.04 安装3090显卡驱动 cuda cudnn pytorch相关推荐

ubuntu20.04 RTX2060||nvidia显卡驱动|cuda|cudnn|pytorch
安装显卡驱动自动安装: ubnutu-drivers devices sudo ubuntu-drivers autoinstall 软件与更新->驱动->安装专有最新版本 cuda11 ...
Ubuntu 20.04 安装NVIDIA显卡驱动+cuda 11.7+cudnn 8.4
Ubuntu 18.04 安装NVIDIA显卡驱动+cuda 10.2+cudnn 本机环境 1 相关查询命令一.Ubuntu 20.04 安装NVIDIA显卡驱动二.Ubuntu 20.04 安 ...
深度学习GPU环境安装教程：Ubuntu16.04+1080(Ti)显卡驱动+CUDA+cuDNN（已n次完美安装）
安装显卡驱动当前只装了ubuntu16.04单系统,亲测可用,之前ubuntu16.04+win10双系统下也是这种方法装的,只是需要切换视频线的接口,可参考这篇 1.装好ubuntu16.04 2 ...
ubuntu 3060显卡驱动+cuda+cudnn+pytorch+pycharm+vscode
文章目录运行环境: 适用: 思路: 1.1 3060显卡驱动自动安装 2.1 CUDA11.1.1 1)下载CUDA Toolkit 11.1 Update 1 Downloads 2)contun ...
ubuntu22.04安装显卡驱动+cuda+cudnn
ubuntu22.04安装显卡驱动+cuda+cudnn 1. 下载驱动和卸载.禁用自带驱动程序 1.1 查看系统显卡型号 1.2 从NVIDIA官网下载相应驱动 1.3 卸载Ubuntu自带的驱动程 ...
Ubuntu16.04安装Nvidia显卡驱动（cuda）
[专业亲测]Ubuntu16.04安装Nvidia显卡驱动(cuda)--解决你的所有困惑确实是可行的下载完成之后: 卸载原先的所有驱动: #for case1: original driver ...
Nvidia3090显卡驱动+CUDA+cuDNN安装步骤
文章目录显卡驱动 CUDA cuDNN 显卡驱动 sudo add-apt-repository ppa:graphics-drivers/ppa # 添加Nidia镜像 sudo apt upda ...
【亲测】Ubuntu16.04手动安装nvidia显卡驱动+CUDA 8.0--Acer E5-572G版
前言前段时间,配置实验室新服务器上的Tesla P4,结果一直有问题,最后终于解决之后.昨天晚上想在自己的笔记本上安装pytorch,sudo apt-get update的时候,结果提示系统缺少一 ...
【nvidia】ubuntu 16.04 安装 gtx1080显卡驱动
安装gtx1080显卡驱动注意:nvidia显卡有个热插拔无法识别问题,如果发现无法显示dell logo可能是线路问题(当前采用的是HDMI转VGA的方式)处理方法是关机->拔电源-> ...

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

安装驱动

安装cuda

安装cudnn

安装pytorch

测试

ubunt16.04 安装3090显卡驱动 cuda cudnn pytorch相关推荐

最新文章

热门文章