Linux命令

切换用户

su username

password

安装pyTorch环境

查看服务器CUDA最高版本

nvidia-smi

可以查看到我的服务器CUDA Version: 11.7

创建conda虚拟环境

conda create -n pytorch python=3.9

进入pytorch环境

conda activate pytorch

安装pytorch

进入官网

根据个人pc或者服务器的配置选择pytorch版本

开始选择了conda安装，但是一直卡在solving environment那里，后来查到可以在anaconda下使用pip3安装，就选择了pip安装，安装的速度反而很快。

pip3 install torch torchvision torchaudio

安装完成

测试安装成功

(pytorch) zy@node01:/home$ python
Python 3.9.16 (main, May 15 2023, 23:46:34) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'2.0.1+cu117'
>>> torch.cuda.is_available()
True
>>> exit()

测试成功