cuda 11.7 从清华源安装pytorch
时间: 2023-07-04 17:08:15 浏览: 564
清华镜像源,用于帮助快速下载pytorch
你可以按照以下步骤从清华源安装 PyTorch:
1. 首先,安装 CUDA 11.7。你可以按照清华镜像源的指南安装 CUDA 11.7:https://mirrors.tuna.tsinghua.edu.cn/help/nvidia-cuda-ubuntu/
2. 接下来,安装 PyTorch。你可以通过以下命令安装 PyTorch:
```
pip3 install torch torchvision torchaudio -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
```
这个命令会从清华镜像源下载 PyTorch。
3. 安装完成后,你可以通过以下命令验证 PyTorch 是否成功安装:
```
python3 -c "import torch; print(torch.__version__)"
```
如果能够输出 PyTorch 的版本号,则说明安装成功。
希望这可以帮助到你!
阅读全文