windows 安装PyTorch 使用清华源
时间: 2023-07-03 15:32:52 浏览: 513
您可以使用清华大学开源软件镜像站来加速PyTorch的下载和安装。按照以下步骤操作:
1. 首先,打开清华大学开源软件镜像站(https://mirrors.tuna.tsinghua.edu.cn/)。
2. 在页面上方的搜索框中输入“PyTorch”并按下回车键。
3. 在搜索结果中找到适合您的PyTorch版本,例如“torch-1.9.0-cp39-cp39-win_amd64.whl”(适用于Python 3.9和Windows 64位操作系统)。
4. 复制PyTorch的下载链接。
5. 打开命令行窗口并运行以下命令来使用清华源安装PyTorch:
```
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple <PyTorch下载链接>
```
将<PyTorch下载链接>替换为您刚刚复制的PyTorch下载链接。
例如,如果您要安装PyTorch 1.9.0版本,可以使用以下命令:
```
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.9.0+cpu -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
如果您的计算机上有NVIDIA的GPU,并且您想要使用GPU来加速模型训练,则可以使用以下命令:
```
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
6. 等待安装完成。
安装完成后,您可以在Python中导入PyTorch并开始使用它。如果您在安装过程中遇到任何问题,请参考官方文档或在PyTorch的GitHub页面上寻求帮助。
阅读全文