CUDA安装

参考文章https://zhuanlan.zhihu.com/p/13191527334

首先查询你的显卡支持最高什么版本的CUDA

打开Nvidia控制面孔

左下角系统信息——组件中查看

 然后去官网上下载CUDA Toolkit 

https://developer.nvidia.com/cuda-downloads

pytorch目前官网给的是支持cuda12.4?

于是找到12.4的下载链接

https://developer.nvidia.com/cuda-12-4-1-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
选择适合自己环境的安装包

 

安装完成后，打开终端或命令提示符，输入nvcc --version，确认CUDA已成功安装。

在Python中 导入pytorch并验证CUDA可行性
import torch

torch.cuda.is_available()

返回True则表示可用