本地安装gpu版本torch(超简单)

1)创建并激活环境

conda create -n DOT1 python==3.9
conda activate DOT1

2)方法1:命令行安装torch

① 查看cuda版本【12.6】

nvidia-smi        # 查看电脑支持的cuda版本

本地安装gpu版本torch(超简单)_第1张图片

nvcc -V        # 查看安装的cuda版本

② 去pytorch官网找命令

pytorch官网PyTorch

本地安装gpu版本torch(超简单)_第2张图片

③ 直接输入图中命令:

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

3)方法2:本地安装

① 输入命令

pip3 install torch

你可能感兴趣的:(python)