安装pytorch时,由于我自己在Ubuntu中安装的cuda,而且也习惯使用pip安装库。遇到pip下载速度太慢。
- 解决方法:临时给pip更换源
- 命令格式:
sudo pip install 包名 -i 镜像源url
- 例如:
pip install torch torchvision -i https://pypi.douban.com/simple/
下面是几个常用的源:
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/
-----------------------------------------------------------------------------
conda更新慢,同样可以换源,一般我采用清华镜像所提供的源
地址:清华Anaconda镜像
代码:
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 --set show_channel_urls yes
此外,可以使用conda检索可按照的库,例如我要安装tensorflow-gpu,首先可以执行
conda search tensorflow-gpu
查看conda中所支持安装tensorflow的哪些版本。
其次选择自己需要的版本安装,
conda install tensorflow-gpu==1.13.1
pip也可以指定版本
pip install tensorflow-gpu==1.13.1