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