安装pytorch,看准版本
nvidia-smi
uname -a
cat /proc/version
找到自己对应的版本,然后直接在官网上找到Start Locally | PyTorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
我在对应的环境里用这个命令,下载也蛮快的
输入python
torch.cuda.get_device_name(0)
输出即正常使用