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则表示可用

__EOF__

本文作者：cheng_zhi
本文链接：https://www.cnblogs.com/chengzhid/p/18662167.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！