Linux服务器环境配置-Pytorch

1.Anaconda安装

利用xshell连接上服务器后首先进行anaconda的安装,使用wget命令进行下载:

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh

下载完之后进行安装:

bash Anaconda3-5.3.1-Linux-x86_64.sh

之后会提示很多,一般输入yes即可,到最后会提示是否安装vscode,根据自己的需求选择即可。到此已经安装完毕,安装完成之后输入conda相关指令时可能会提示:conda: command not found,此时代表初始化未成功,输入指令:source anaconda3/bin/activate即可进入base环境,然后输入conda deactivate退出base环境就可以正常使用。

但由于安装的anaconda版本过低,可以先进行更新,更新指令如下:

conda update -n base -c defaults conda

 2.Pytorch安装

首先先创建一个环境:

conda create --name torch_1_8 python=3.8

--name后面为环境名,可以随便取,创建环境之后可以进入环境:

conda activate torch_1_8

然后通过nvidia-smi查看自己服务器的cuda版本,在pytorch官网上下载对应版本的torch(这里不推荐换清华源,换清华源可能会出现下载完版本为cpu版本的情况),输入命令即可下载:

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

到此应该就安装完毕了,验证安装是否成功通过以下命令即可判断:

Linux服务器环境配置-Pytorch_第1张图片 

 

你可能感兴趣的:(linux,pytorch,python)