目录
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
至此,配置完毕。