烟台官网首页_网络系统架构图_宁德市人民政府_郑州专业seo推荐

安装命令选择

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)