pytorch安装包国内镜像源的命令
时间: 2024-12-19 18:26:05 浏览: 10
PyTorch离线安装包+官网源/清华源
PyTorch在国内安装时,可以使用阿里云、清华大学等提供的镜像源来加速下载过程。以下是安装PyTorch及其依赖的一些常用命令,通常会先添加一个国内的pip源:
```bash
# 添加阿里云的PyTorch源
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
# 或者清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/
```
然后你可以通过`pip install torch torchvision`来安装PyTorch和相关库,如果需要显卡支持,记得加上`--cuda-version`指定CUDA版本,如`pip install torch torchvision --cuda=10.2`.
注意,如果你是在Anaconda环境中,也可以在创建新环境时就设置镜像源,例如:
```bash
conda create -n myenv python=3.8 -c anaconda -c mingw -c nvidia
conda activate myenv
conda config --add channels https://mirrors.aliyun.com/anaconda/pkgs/free/
conda config --add channels https://mirrors.aliyun.com/anaconda/pkgs/r/
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
阅读全文