ubuntu 安装cuda和cudnn

还在为安装cuda和cudnn烦恼吗？教你一键安装所需的环境。

首先你本机的安装一个conda，不知道怎么安装conda环境，可以去我主页搜索conda，照步骤操作；

然后激活你创建的虚拟环境，准备安装所需环境；

进入网页：PyTorch

选择你需要的一系列东西，如：

使用Run this Command命名

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

创立所需的环境，这样就可以了；

cudnn 安装

虚拟环境安装：

conda install cudnn

直接下载安装：参考链接：Ubuntu16.04下安装cuda和cudnn的三种方法（亲测全部有效）_老王回归的博客-CSDN博客_ubuntu安装cuda

查看cuda是否安装成功的命名行：

import torch
print(torch.cuda.current_device())   # 返回当前设备索引
print(torch.cuda.device_count())    # 返回GPU的数量
print(torch.cuda.get_device_name(0))   # 返回gpu名字，设备索引默认从0开始
print(torch.cuda.is_available())   # cuda是否可用
print(torch.version.cuda)  # cuda版本

自己操作：

使用conda list查看：

参考：完美解决（最简单）：不同虚拟环境不同CUDA版本的安装问题（tensorflow-gpu与pytorch）_TC1398的博客-CSDN博客_虚拟环境安装cuda