最近总是换服务器,搭建了N次pytorch环境,决定总结一下经验。
①安装anaconda
从官网(https://www.anaconda.com/download/#linux)下载安装文件,上传到服务器,使用bash指令进行安装:
$ bash Anaconda3-5.0.1-Linux-x86_64.sh
不想看licensewen文件可以按q跳过,注意安装完成后,程序会问是否把anaconda的binary路径加入到.bashrc,建议选择yes,这样之后在终端可以直接用python、spyderd使用当前anaconda配置的环境,也可以使用anaconda-navigator指令。
注意选择yes之后并不会立即生效,可以使用指令
$ source ~/.bashrc
后生效。
②安装pytorch、torchvision
查看当前服务器使用Ubuntu、cuda、python版本,然后在pytorch官网(http://pytorch.org/)下载对应的pytorch whl文件。
查看cuda版本的方法:
$ cat /usr/local/cuda/version.txt
然后在whl文件所在的目录下使用pip指令安装:
$ pip install torch-0.3.1-cp36-cp36m-linux_x86_64.whl
安装完成后再python中检测是否成功安装的方法:
import torch
torch.cuda.is_available()
再安装torchvision:
$ pip install torchvision
以及一些常用的包:
$ pip install gpustat
$ pip uninstall pillow
$ pip install pillow-simd
$ pip install scikit-learn
$ pip install opencv-python
$ pip install matplotlib
其他tips:
为当前用户分配某个文件夹的权限:
$ sudo chown -R tfan /media/sdd/cjc