ubuntu服务器pytorch环境搭建

最近总是换服务器,搭建了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

【ubuntu服务器pytorch环境搭建】

    推荐阅读