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环境搭建】
推荐阅读
- pytorch|使用pytorch从头实现多层LSTM
- 探索免费开源服务器tomcat的魅力
- [源码解析]|[源码解析] NVIDIA HugeCTR,GPU版本参数服务器---(3)
- Pytorch|Pytorch AlexNet Fashion-MNIST
- VM|VM ware 的 harbor 私有仓库搭建 (Ubuntu16.04)
- pytorch|YOLOX 阅读笔记
- Xshell5|Xshell5 远程连接本地虚拟机Ubuntu16
- Pytorch学习|sklearn-SVM 模型保存、交叉验证与网格搜索
- mac|mac 链接linux服务器 如何在Mac上连接服务器
- Linux|Linux 服务器nginx相关命令