1. 安装环境

1.1. 安装pytorch环境

1.1.1. 下载

下载地址，选择好后直接执行命令即可。

1.2. cuda环境

1. 因为是 WSL 环境，所以无需安装显卡驱动
2. 下载 cuda-toolkit

1. 访问网站 cuda-toolkit-archive 下载对应的 cuda 版本并安装

2. 安装后配置环境变量

sudo vim /etc/profile

export CUDA_PATH=/usr/local/cuda-11.8
export PATH=${CUDA_PATH}/bin:${PATH}
export LD_LIBRARY_PATH=${CUDA_PATH}/lib64

3. 下载并安装 cudnn，下载地址 cudnn, 登录可以使用微信登陆即可, 下载完直接执行 dpkg 安装即可

3. 验证 cuda 安装成功
4. 查看 cuda 安装版本

nvcc -V

1.3. 代码中测试 cuda 环境是否可用

import torch

print("cuda 设备数量: " + str(torch.cuda.device_count()))
print("cuda 是否可用: " + str(torch.cuda.is_available()))
print("cuda 当前设备: " + str(torch.cuda.current_device()))