本地装了个pytorch cuda

安装命令选择

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

torch版本查看

python
import torch
print(torch.__version__)

 查看pytorch能否使用cuda

import torch
 
# 检查CUDA是否可用
print(torch.cuda.is_available())
 
# 显示当前CUDA版本
print(torch.version.cuda)