旅游网站内容规划特点,wordpress能做成社区吗,网站编程软件有哪些,临沂企业网站建设解决思路#xff1a;import torch正常#xff0c;说明torch包安装正常#xff0c;但是不能和gpu正常互动#xff0c;猜测还是pytroch和cuda的配合问题
1.查看torch包所需的cuda版本
我的torch是2.0.1#xff0c;在现在是比较新的包#xff0c;需要12以上的cuda支持import torch正常说明torch包安装正常但是不能和gpu正常互动猜测还是pytroch和cuda的配合问题
1.查看torch包所需的cuda版本
我的torch是2.0.1在现在是比较新的包需要12以上的cuda支持我用nvcc -V或者set cuda查看当前环境路径下的cuda版本显示为11.1这显然无法匹配
2.匹配cuda版本
那就需要更新了首先检查我的硬件是否支持新版本的cudanvidia-smi可以看到我的显卡最高支持到12.3因此就去官网下载CUDA Toolkit 12.3 Update 2 下载 |NVIDIA 开发者
但是安装前这时候还要看cuda和geforce驱动匹配问题 3.匹配geforce驱动
cuda和geforce驱动的匹配关系可以看到如果想安装cuda12.1至少匹配到527.41
CUDA 12.3 Update 2 Release Notes (nvidia.com) 进入geforce驱动官网
NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA 安装Pytorch时NVIDIA驱动更新CUDA版本问题_为什么pytroch不更新amd驱动-CSDN博客 4.开始安装cuda 这网页进来 version没有12 你别管 直接看大标题对就下
CUDA Toolkit Archive | NVIDIA Developer