安装cuda与pytorch

cuda安装：https://blog.csdn.net/AI_dataloads/article/details/133043869
查看gpu算力：https://developer.nvidia.com/cuda-gpus

首先卸载当前安装的pytorch：

pip uninstall torch torchvision torchaudio

安装pytorch：https://pytorch.org/get-started/locally/
注意别装了cpu版本的，只要你的cuda version高于pytorch的就行

在cmd窗口查看gpu信息：nvidia-smi

安装完成后在vscode确认：

import torch


print("PyTorch version:", torch.__version__)  # 检查 PyTorch 版本
print("CUDA available:", torch.cuda.is_available())  # 检查 CUDA 可用性
print("CUDA version:", torch.version.cuda)  # 检查 PyTorch 使用的 CUDA 版本
print("Number of GPUs:", torch.cuda.device_count())  # 检查可用的 GPU 数量

'''
PyTorch version: 2.4.1+cu124
CUDA available: True
CUDA version: 12.4
Number of GPUs: 1
'''

注意安装完成后可能需要更新驱动，重启电脑之类的操作