Ubuntu22.04.4系统/安装python3.9/pytorch/torchvision【GPU版】

1.安装python3.9

1.1 创建python3.9的虚拟环境

conda create -n QwenChat python=3.9

1.2 输入“y”

1.3 创建成功

2.安装pytorch和torchvision

2.1 进入虚拟环境

进入刚刚创建的虚拟环境

conda activate QwenChat

2.2 conda安装

• 查看cuda的版本

• 浏览器打开网址PyTorch
• 鼠标往下翻，到参数选择界面（因无CUDA12.2，选择了最接近的CUDA12.1），复制【Run this Command】的命令

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

• 运行【Run this Command】的命令

• 输入“y”后，稍等片刻，安装完成

2.3 测试

python -c "import torch; print(torch.__version__)"

python -c "import torchvision; print(torchvision.__version__)"