安装CUDA和CUDNN

安装CUDA

文件已经上传至百度网盘

链接：https://pan.baidu.com/s/1LKzZXtSr1kXOnlfbO0cmEw?pwd=gfbb 
提取码：gfbb

sudo sh cuda_11.3.1_465.19.01_linux.run

继续

在接受许可之后，不选择驱动，然后选择安装

安装CUDNN

解压下载的cudnn
将库文件复制到对应的cuda目录

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

下面语句检查安装成功，能否出现CUDA版本

nvcc-V

安装Anaconda3

bash Anaconda3-2023.03-1-Linux-x86_64.sh

1、提示同意许可协议，输入yes
2、还要选择安装路径，可自定义还可选择默认的
3、可能再问你询问Do you wish the installer to initialize Anaconda3 by running conda init?
直接yes
！！！安装好的话，就会在新窗口自动出现base

安装Pytorch

1. 创建虚拟环境：python版本为X.X、名字为your_env_name

conda create -n pytorch1.12 python=3.9

2. 激活虚拟环境

conda activate pytorch1.12

3. 查看conda的环境

conda info -e

4. 激活虚拟环境

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

5. 测试

python
import torch
torch.__version__
torch.cuda.is_available()