(环境搭建五-简介版) ubuntu下安装gpu版本的pytorch

背景：需要配置一个2张3080ti配置新机器的深度学习环境

一 安装anaconda

官网：

Anaconda | Individual Edition

 选择第一个

二 安装或查看显卡驱动

 1 官网下载驱动并安装，NVIDIA 驱动程序下载

我的电脑已经装了驱动，

2 看下自己的显卡型号和驱动，（貌似电脑出厂已经安装了驱动）

nvidia-smi

显卡被省略
驱动是495.29.05

3 安装Anaconda

这个比windows简单，一条命令。

首先切换到anaconda安装包的路径，在该路径下执行指令(注意sh文件名字要写自己的)

sh Anaconda3-5.0.1-Linux-x86_64.sh   #后边的文件名称是你的安装包的名称

输入python3 查看是否安装成功，上述表示成功

三 安装驱动对应的cuda

cuda官网 CUDA 工具包 | NVIDIA Developer

CUDA Toolkit 11.5 Downloads | NVIDIA Developer

下载cuda,  新需要与系统对应，我的Ubuntu 20.04.3 LTS

 因此选择20.04

查看自己cuda版本

11.5

四 安装cudnn

1 cuda与cudnn是对应的，看好对应关系

表太旧

查看cudnn的版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

输入之后，没结果说明没安装

2 官网

https://developer.nvidia.com/rdp/cudnn-downloadhttps://developer.nvidia.com/rdp/cudnn-download

去官网下载cudnn选择对应版本

 选择x86 64

但是找不到连接

下载不了，粘贴到下载器可以下载

https://developer.download.nvidia.com/compute/machine-learning/cudnn/secure/8.2.4/11.4_20210831/cudnn-11.4-linux-aarch64sbsa-v8.2.4.15.tgz?DW4R9WomgN68tDH-dl2UKUO8OIFpbumcxvd366AnroTQH1dZ_ZknKLQr6QUM6cqko3kz6aCClrmnB7yKwCEXzpQWXNedHW7-jjLGPksD39N3BZlEAtNWQQRjklcAunXGrFdqTIFr1m5idrl22WtTRkjcX6B4YoZVWRTSLV1A-qxWyew5dhc1ZTSGZMlul5OL1QEpdQDQnyz06ZmNmZzVMVC3sA&t=eyJscyI6InJlZiIsImxzZCI6IlJFRi13d3cuYmFpZHUuY29tXC9saW5rP3VybD1RRWlUV2dOTDBBd1hhR0NTQm51TjJVMUlfVS1hNG1GamN0NHBaU3V3a3FnbE5SZ3RIazR1ZEw5cHNiRk15c0FycXpZNEhhSW5xMjFoYnUwNnpsOHdSSyZ3ZD0mZXFpZD1iZmU1ZGExMTAwMDQ5MWVmMDAwMDAwMDM2MTk3MTgyYiJ9