Skip to content

Latest commit

 

History

History
22 lines (22 loc) · 1.67 KB

解决pytorch安装失败.md

File metadata and controls

22 lines (22 loc) · 1.67 KB

解决pytorch安装失败

因为国内网络问题,即便anaconda配制了清华源,有时候安装pytorch gpu版本的时候还会失败.从安装pytorch1.7.0 gpu版本的实际情况看,有时按 pytorch官方文档的命令装一两天都装不上.这里总结了一个步骤,虽然也比较慢,但最终还是可以装好.也算是一个在较差网络环境下的"最佳实践".

  1. 在xx云上买一个国外的云主机.
  2. 在云主机上装miniconda. 大家可能更熟悉anaconda,miniconda和anaconda相比只保留了python环境创建,下载第三方包功能, 去掉了jupter notebook的功能.
  3. 用国外云主机的miniconda安装gpu版本的pytorch,命令如下:
    conda create -n pytorch1.7.0 python=3.8.5
    conda activate pytorch1.7.0
    conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
    
    我买的云主机在德国,十几分钟就安装好了,再也不用配制conda的源了.
  4. pytorch被安装到了$CONDA_HOME/envs/pytorch1.7.0目录下. 把pytorch1.7.0目录打包成压缩包,压缩包有2G多.
  5. 用scp命令把压缩包拖到本地,这个过程比较漫长,我大概弄了两三个小时. 成功后解压到本地的$CONDA_HOME/envs/pytorch1.7.0目录.
  6. 修改$CONDA_HOME/envs/pytorch1.7.0/bin/pip文件,把第1行python的位置修改成和本地机一致(之前是云主机的).之后就可以按 以前的方法使用pytorch1.7.0环境了.
  7. 比较绕的方法,没有办法的办法.没有办法情况下的“最佳实践”

参考资料

pytorch官方文档
conda国内源官网
miniconda官网