cuda11.2安装pytorch用清华镜像源
时间: 2023-10-01 12:04:30 浏览: 494
清华镜像源,用于帮助快速下载pytorch
可以参考以下步骤:
1. 安装清华镜像源:在终端中输入以下命令:
```shell
sudo cp /etc/apt/sources.list /etc/apt/sources.list.backup
sudo sed -i 's/archive.ubuntu.com/mirrors.tuna.tsinghua.edu.cn/g' /etc/apt/sources.list
sudo apt-get update
```
2. 安装 CUDA 11.2:可以从 NVIDIA 的官网下载安装包,也可以通过以下命令安装:
```shell
sudo apt-get install nvidia-cuda-toolkit
```
3. 安装 PyTorch:可以通过 pip 安装,命令如下:
```shell
pip install torch torchvision torchaudio -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
```
希望这些步骤可以帮助你成功安装 PyTorch!如果还有其他问题,请随时提出。
阅读全文