清华镜像安装pytorch-gpu
时间: 2024-12-11 14:16:34 浏览: 37
pytorch安装GPU版本cuda和cudnn安装配置
在清华大学源(也称为`mingguo`镜像)上安装PyTorch-GPU,你需要先确保已经配置了相应的清华源环境。以下是通过pip安装的基本步骤:
1. **更新pip**:
```
pip install --upgrade pip
```
2. **添加清华源**:
对于Python 3.6及更高版本,你可以运行:
```
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/
```
或者直接设置系统级别的pip源,例如在Linux中:
```bash
echo "export PYTHON_GET_PYPI_URL=https://pypi.tuna.tsinghua.edu.cn/simple/" >> ~/.bashrc
source ~/.bashrc
```
3. **安装PyTorch-GPU**:
使用以下命令安装特定版本的PyTorch,如果需要CUDA支持,记得指定合适的GPU版本:
```
pip install torch torchvision -f https://download.pytorch.org/whl/torch_stable.html --index-url https://pypi.tuna.tsinghua.edu.cn/simple/
```
如果有多个版本选择,可以加上`--extra-index-url`指定额外的索引。
4. **检查安装**:
安装完成后,你可以通过`import torch`来验证是否成功安装了PyTorch及其GPU版本。
阅读全文