pip或conda安装pytorch
缘由
最近参加学校老师的深度学习实习,接触到在服务器上安装pytorch,网上教程很多,但比较乱,所以自己总结了一下(以后可以备用)。
两种方法:第一使用Anaconda自带的 pip 进行安装,第二使用 conda 安装 。(建议使用第一种,亲测最快)
第一种:使用pip安装
首先,进入服务器终端,首先查看当前pip是否是Anaconda自带,(如果不是,后期会需要修改好多环境变量)查看方法:直接在终端输入
~$ pip -V
结果如下:
然后,在pytorch官网( https://pytorch.org/get-started/locally/ )上选择好自己的配置会自动生成 默认安装命令,如下图所示:
但是上面这个命令安装一般会非常非常慢 ,可以通过修改pytorch的源 来加速,这里使用清华源,命令如下:
~$ pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/
#这里使用的是pip,根据自己Anconda决定
#安装过程如果出错,可以多试几遍
安装完成后,可以使用python调用实验,如下:
第二种,使用conda安装:
官网默认命令行:
添加清华源 提速:
#添加清华源
conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
安装完成后,可以使用python验证。