利用清华镜像安装cuda12.1对应的pytorch1.9.0
时间: 2023-07-25 17:08:45 浏览: 193
好的,安装步骤如下:
1. 首先在清华镜像站中找到对应的CUDA安装文件,选择合适的版本进行下载。
2. 下载完成后,解压缩文件并进入安装目录,执行以下命令安装CUDA:
```
sudo sh cuda_12.1.0_460.32.03_linux.run
```
安装过程中需要输入一些选项,可以根据自己的需求进行选择。
3. 安装完成后,配置环境变量。在`~/.bashrc`文件中添加以下内容:
```
export PATH=/usr/local/cuda-12.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH
```
然后执行以下命令使其生效:
```
source ~/.bashrc
```
4. 接下来安装PyTorch。在终端中执行以下命令安装PyTorch:
```
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
如果需要安装其他版本的PyTorch,只需要修改命令中的版本号即可。
5. 安装完成后,可以在Python中导入PyTorch模块进行测试,例如:
```
import torch
print(torch.__version__)
```
如果输出的版本号是1.9.0,则说明安装成功。
希望这个步骤能够帮助到你。
阅读全文