清华镜像安装gpu版本的pytorch
时间: 2023-10-28 16:57:17 浏览: 250
pytorch gpu版
以下是在清华镜像安装GPU版本的PyTorch的步骤:
1. 打开清华镜像站的PyTorch页面:https://mirrors.tuna.tsinghua.edu.cn/help/pytorch/
2. 找到适合你的CUDA版本的PyTorch whl包,比如我选择CUDA 10.2的pytorch 1.7.0版本,就可以找到这个whl文件:https://mirrors.tuna.tsinghua.edu.cn/pytorch/whl/cu102/torch-1.7.0%2Bcu102-cp37-cp37m-linux_x86_64.whl
3. 打开终端,输入以下命令安装PyTorch:
```
pip install https://mirrors.tuna.tsinghua.edu.cn/pytorch/whl/cu102/torch-1.7.0%2Bcu102-cp37-cp37m-linux_x86_64.whl
```
注意:以上命令是安装CUDA 10.2版本的PyTorch,如果你的CUDA版本不同,请修改命令中的cu102为你的CUDA版本,比如cu101。
4. 安装完成后,你可以在Python中使用import torch来验证PyTorch是否已经成功安装。
希望以上步骤能够帮助到你。
阅读全文