一：设备信息

查看显存配置

shell中输入：

nvidia-smi

工具：vscode

python：3.12

驱动：cuda12.6

显卡：RTX4060

二：windows系统

1.创建虚拟环境

# 在D盘创建env文件夹，用来管理虚拟环境
mkdir D:\envs  -Force   

2.创建项目环境pytorch

conda create --prefix D:\envs\pytorch python=3.12 -y

3.激活环境

conda activate D:\envs\pytorch

打开编辑器vscode

4.安装pytorch库

这里安装的是 GPU 版的 PyTorch（支持 CUDA 12.6，RTX 4060）

注意：读者可以根据自己的驱动与GPU去选择相应的版本

python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

5.检测GPU是否可用

编辑器终端执行：

python -c "import torch; print(torch.__version__); print(torch.version.cuda); print(torch.cuda.is_available()); print(torch.cuda.get_device_name(0) if torch.cuda.is_available() else '')"

三：linux系统

1. Anaconda下载及初始化

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh

chmod +x Miniconda3-latest-Linux-x86_64.sh

./Miniconda3-latest-Linux-x86_64.sh -b

~/miniconda3/bin/conda init

source ~/.bashrc

2.同意服务条款

conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main
conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r

3.创建虚拟环境

# 清除缓存
conda clean -i

conda create -n pytorch python=3.12 -y

4.安装pytorch

同理