最近需要在linux服务器上安装pytorch,先用anaconda创建了环境,但是在用conda安装pytorch的时候特别慢。最开始已经添加了清华源,但是使用如下命令安装pytorch的时候一直卡在solving environment不动,按照网上说的已经去掉了-c pytorch选项,且开了科学上网,都不好使。
然后发现,用pip加上-i可以很快地安装好(注意如果是python2的话不能用pip3而要用pip)
pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package
如果你之前已经用conda创建了pytorch环境:
conda create -n torch python=3.6 numpy matplotlib pandas anaconda
如果想要把pytorch安在这个环境下而不是base环境下,只需要先激活环境(torch是你自己取的这个环境的名字)
conda activate torch
然后再使用上面的pip命令安装即可,不必用conda安装。也就是说,在anaconda环境下用pip安装也是可以的,只要你先激活进入你想安装pytorch的环境。
pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package
成功!