pycharm清华源可以下载gpu的pytorch吗
时间: 2023-12-26 09:04:03 浏览: 149
pytorch换源下载
可以的。你可以在PyCharm中使用清华源下载GPU版本的PyTorch。具体步骤如下:
1. 打开PyCharm,创建一个新的项目。
2. 在PyCharm中打开终端。
3. 运行以下命令,将清华源添加到PyTorch的源列表中:
```
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
```
4. 运行以下命令,安装GPU版本的PyTorch:
```
pip install torch torchvision -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
其中,cu111代表CUDA 11.1版本,如果你的CUDA版本不同,请替换为对应的版本号。
5. 等待安装完成后,你就可以在PyCharm中使用GPU版本的PyTorch了。
阅读全文