加速Pytorch下载速度(pip)
如果使用官网提供的pip时,速度一般在300K左右,Pytorch大小在2G+以上,需要想办法加速
1、配置本地的pip镜像
windows10:
在当前用户下,新建一个目录 pip 比如
C:\Users\admin\pip
然后新建一个文件 pip.ini
内容如下:
[global]
index-url = https://pypi.douban.com/simple/
2、在pytorch网站获得内容安装命令,如gpu方式:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
这个时候在命令行会下载,速度比较慢,如下:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
Looking in indexes: https://pypi.douban.com/simple/, https://download.pytorch.org/whl/cu116
Collecting torch
Downloading https://download.pytorch.org/whl/cu116/torch-1.12.1%2Bcu116-cp39-cp39-win_amd64.whl (2388.0 MB)
---------------------------------------- 0.0/2.4 GB 213.6 kB/s eta 3:06:10
给了路径后,直接通过迅雷下载,速度可以到10M以上,下载后,直接安装
3、直接安装
pip3 install torch-1.12.1+cu116-cp39-cp39-win_amd64.whl
安装完毕后,再采用命令安装
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
这个时候安装只要安装 torchvision 和 torchaudio,速度提升不少