安装cuda
参考博客 https://blog.youkuaiyun.com/weixin_42795611/article/details/93065567 写的很详细
安装anaconda
首先下载anaconda,由于官网的下载速度十分缓慢,而且还不支持断点续传,经常下载失败。故而,去国内的镜像站点下载,
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
选择对应的版本下载,下载完成后上传到linux机器上去,服务器选择filezilla上传,虚拟机直接拖拽进去就ok了。
然后打开终端,输入:
bash Anaconda3-5.2.0-Linux-x86_64.sh
一路回车+输入yes即可。
完成安装后,关闭终端,并重新打开一个终端,输入conda:
则安装完成,输入:sudo gedit ~/.bashrc
sudo gedit ~/.bashrc
检查末尾是否已经添加如下变量,没有则添加:
2.创建虚拟环境
conda create -n name python=3.5 (name和python版本号可以更改)
3. 进入虚拟环境
source activate name
4. 安装pytorch
conda install pytorch=0.4.0 torchvision cudatoolkit=9.0(安装指定版本的pytorch)
注:如有需要可以将默认的源换成国内的清华源,由于一些原因默认的源国内下载速度过慢,在终端输入以下:
//这个是清华pytorch源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
查看修改是否成功,输入:
cat ~/.condarc
安装好后可通过 conda info --envs 命令查看当前已安装的pythorch个数和名称