1. 查看服务器的操作系统

执行代码：

uname -a

发现是Ubuntu 22.04

2. 查看服务器的cuda版本

执行代码：

nvcc -V

发现是cuda12.6

3. 配置开发环境

ubuntu22.04+miniconda3+Python3.10+pytorch

3.1 安装miniconda3环境

执行代码：

# 下载并安装 miniconda3
mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm -rf ~/miniconda3/miniconda.sh 

安装完miniconda3后还需要初始化，执行代码：

~/miniconda3/bin/conda init bash
~/miniconda3/bin/conda init zsh

3.2 创建虚拟环境

执行代码：

source ~/.bashrc
conda create --name myenv python=3.10

3.3 在虚拟环境中安装pytorch

pytorch官网：https://pytorch.org/get-started/locally/，选择合适的版本。

执行代码：

conda activate myenv # 进入虚拟环境
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

验证是否安装成功，执行代码：

python
>> import torch
>> print(torch.__version__) # 如果安装成功，这里会显示版本

3.4 在虚拟环境中安装库

执行代码：

conda install xxx

至此，配置完毕。