您的位置:首页 > 其它

ubuntu服务器pytorch环境搭建

2018-03-03 11:51 621 查看
最近总是换服务器,搭建了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

其他tips:
为当前用户分配某个文件夹的权限:
$ sudo chown -R tfan /media/sdd/cjc
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: