pytorch安装教程清华源头
时间: 2023-11-12 18:57:26 浏览: 89
清华源头的PyTorch安装教程如下:
1. 首先安装Anaconda或Miniconda,可以从官网下载对应的安装包进行安装。
2. 创建一个新的conda环境,可以使用以下命令:
```
conda create -n pytorch python=3.7
```
3. 激活conda环境,可以使用以下命令:
```
conda activate pytorch
```
4. 添加清华源头的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/
```
5. 安装PyTorch,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1
```
6. 安装完成后,可以使用以下命令验证PyTorch是否安装成功:
```
python -c "import torch; print(torch.__version__)"
```
阅读全文