Linux服务器安装Anaconda\Pytorch\Tensorflow-gpu

好久都没有写博客了,恰巧在外实习,要在公司的AI服务器上安装anaconda,索性记录下过程。

我使用的shell客户端为mobaxterm,首先安装anaconda:


1、连接服务器,左上角session,然后ssh,输入IP地址和账号名

Linux服务器安装Anaconda\Pytorch\Tensorflow-gpu_第1张图片

2、进入后,cd到安装在路径下

3、安装anaconda,进入anaconda官网https://www.anaconda.com/products/individual,我们使用linux版本,这里安装linux x86版本

Linux服务器安装Anaconda\Pytorch\Tensorflow-gpu_第2张图片

在x86版本的链接上用鼠标右键得到下载地址https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh,使用wget命令下载安装anaconda包

wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh

4、对下载好的anaconda安装包,进行安装

bash Anaconda3-2021.05-Linux-x86_64.sh

一路yes即可(我这里是输入yes,然后更改了下安装的路径,从home改到了data,不更改路径直接忽略即可)

Linux服务器安装Anaconda\Pytorch\Tensorflow-gpu_第3张图片

推荐一篇知乎文章,很详细的解释了anaconda安装过程

ssh 连接 Linux 服务器并安装 Anaconda


安装pytorch

1、创建虚拟环境,环境名字叫torch,Python版本为3.6

conda create -n torch python=3.6

然后激活虚拟环境

conda activate torch

2、此时已经在虚拟环境中了,查看cuda版本,输入

nvidia-smi

可以看到服务器cuda version是11.2(5张卡的服务器,美滋滋,公司还是壕啊)

Linux服务器安装Anaconda\Pytorch\Tensorflow-gpu_第4张图片

3、pytorch官网https://pytorch.org/,找到对应版本,复制命令下载

Linux服务器安装Anaconda\Pytorch\Tensorflow-gpu_第5张图片

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

4、安装完毕后,输入python,然后分别输入以下语句进行测试

import torch
x = torch.rand(5, 3)
print(x)

输出类似语句则安装成功!

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

 

你可能感兴趣的:(linux,anaconda,pytorch,服务器)