anaconda 安装清华园pytorch gpu版
时间: 2023-10-05 09:10:35 浏览: 162
您可以按照以下步骤在清华镜像源下安装 PyTorch GPU 版本:
1. 打开 Anaconda Prompt,输入以下命令创建一个新的虚拟环境:
```
conda create -n pytorch_gpu python=3.8
```
2. 激活虚拟环境:
```
conda activate pytorch_gpu
```
3. 添加 PyTorch 的清华镜像源:
```
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
```
4. 安装 PyTorch GPU 版本:
```
conda install pytorch-gpu torchvision torchaudio cudatoolkit=11.0
```
5. 验证 PyTorch 是否成功安装:
```
python -c "import torch; print(torch.__version__)"
```
如果输出 PyTorch 版本号,则说明安装成功。
阅读全文