ubuntu20.04安装cuda和cudnn

以下安装的版本

1、英伟达驱动:515.65.01

2、CUDA:11.0.2

3、CUDNN: 8.0.5.29

4、pytorch安装的方式,在官网选择命令行输入

conda install pytorch torchvision torchaudio cudatoolkit=11.3

下载英伟达驱动

ubuntu20.04安装cuda和cudnn_第1张图片

(43条消息) Ubuntu20.04安装英伟达驱动步骤_Rootzzy1216的博客-CSDN博客

下载cuda

1、在官网查看合适的版本

我此处下载11.0.2版本

2、按照下面教程下载

(42条消息) Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程(图文)_嵌入式技术的博客-CSDN博客_ubuntu安装cudnn

其中第一步下载依赖,可能会有几个库下载不了,没关系,继续按后面的步骤来

不要参考其中的测试是否安装好的步骤

3、测试,参照下面的教程

(42条消息) cudn/cuDNN/显卡/驱动相关查询_花花少年的博客-CSDN博客_cudnn版本查看

下载cudnn

进入英伟达官网之前需要注册账号

下载的版本 8.0.5

具体参考:(42条消息) Ubuntu20.04安装CUDA11.0和cuDNN8.0_NiuXL1996的博客-CSDN博客_ubuntu20.04安装cuda和cudnn

下载anaconda、pycharm和创建pytorch环境

参考(43条消息) Ubuntu20.04安装Pytorch:Anaconda3+Pycharm+Pytorch+GPU_咸咸咸虾的博客-CSDN博客

卸载pytorch的方式

进入pytorch环境

(58条消息) Pytorch如何更新版本与卸载,使用pip,conda更新卸载Pytorch_醉雨轩^_^的博客-CSDN博客_卸载pytorch

卸载时可以使用pip和conda都试一下

你可能感兴趣的:(电脑配置,ubuntu,linux,运维)