利用清华镜像安装cuda12.1 对应的 PyTorch 版本是 PyTorch 1.9.0。
时间: 2024-06-07 19:11:59 浏览: 445
安装 PyTorch 1.9.0 需要先安装 CUDA 12.1 和 cuDNN 8.2,可以按照以下步骤进行安装:
1. 安装 CUDA 12.1
首先,下载 CUDA 12.1 安装包,可以在清华镜像站上下载,链接为:https://mirrors.tuna.tsinghua.edu.cn/nvidia/cuda/。选择对应操作系统的版本和安装方式,下载后进行安装。
安装完成后,将 CUDA 的 bin 目录添加到系统环境变量中,以便后续使用:
```
export PATH=/usr/local/cuda-12.1/bin:$PATH
```
2. 安装 cuDNN 8.2
cuDNN 是 CUDA 的深度学习库,PyTorch 需要依赖它。同样可以在清华镜像站上下载,链接为:https://mirrors.tuna.tsinghua.edu.cn/nvidia/cudnn/。选择对应的版本进行下载,并按照官方文档进行安装。
安装完成后,将 cuDNN 的库文件路径添加到系统环境变量中:
```
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:/usr/local/cudnn-8.2/lib64:$LD_LIBRARY_PATH
```
3. 安装 PyTorch 1.9.0
最后,可以使用 pip 安装 PyTorch 1.9.0:
```
pip install torch torchvision torchaudio
```
安装完成后,可以在 Python 中导入 PyTorch:
```
import torch
```
如果没有报错,则说明 PyTorch 安装成功。
阅读全文