以下安装的版本
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
下载英伟达驱动
(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
deb方式下载cudnn:Ubuntu22.04安装CUDA、cudnn详细步骤_ubuntu安装cuda和cudnn_takedachia的博客-CSDN博客
下载anaconda、pycharm和创建pytorch环境
参考(43条消息) Ubuntu20.04安装Pytorch:Anaconda3+Pycharm+Pytorch+GPU_咸咸咸虾的博客-CSDN博客
卸载pytorch的方式
进入pytorch环境
(58条消息) Pytorch如何更新版本与卸载,使用pip,conda更新卸载Pytorch_醉雨轩^_^的博客-CSDN博客_卸载pytorch
卸载时可以使用pip和conda都试一下