科研训练6：服务器GPU下创建python环境并根据CUDA版本安装torch包

本博文记录如何在服务器GPU下，创建python环境，并根据CUDA版本安装torch包

Step1:登陆服务器，查看gpu是否可用，并确定CUDA版本

首先用ssh命令登录服务器（账号密码略），输入命令：nvidia-smi，查看gpu是否可用。可以看到这里有2个GPU，CUDA版本是12.0。

Step2:创建python环境，并激活

（如果没有conda要先安装anaconda）输入如下命令，含义是创建一个name是pytorch_law的环境，python环境是3.9：

conda create -n pytorch_law python=3.9

一些有关环境常用的命令：

展示所有环境：conda env list；切换环境：source activate python_env；退出环境：source deactivate python_env；卸载环境：conda remove --name python_env --all。此处激活这个环境：

conda activate pytorch_law

Step3:查看pytorch安装版本

登陆pytorch官网（https://pytorch.org），选择linux系统，pip或者conda安装，语言是python，选择cuda12.1（不确定版本可以再页面上：install previous version of pytorch查看是否兼容），然后下方会呈现命令：pip3 install torch torchvision torchaudio（每个选项下是不一样的）。

注：如果pip下载比较慢，可以pip换源换成清华源，然后再用pip安装

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

 Step4：验证是否顺利安装

输入python进入命令行，再输入import torch，如果没有报错，就是顺利安装！