win10+GeForce 940mx安装cuda | 您所在的位置:网站首页 › 940m驱动 › win10+GeForce 940mx安装cuda |
起因是在python3里执行下述代码,发现是false,于是想想怎么解决。 import torch torch.cuda.is_available()首先搜到在windows中安装cuda的指引,发现主要步骤为: 1. 配平显卡驱动、cuda版本、pytorch版本。我是pytorch 1.11.0装好了才发现这个问题,根据前车之鉴,我选择把显卡驱动升级为最新版本512.95,并开始准备安装1.11.0的cuda。 2023年4月24日:补充具体步骤 1. 在Previous PyTorch Versions | PyTorch网址中找自己已经装了的pytorch版本号(如1.8.0) 注意,如果之前已经安装了(像我就错装了cpu版本的torch),必须要先卸载(最好pip和conda都uninstall一下),不然就会遇到问题:Key already registered with the same priority: GroupSpatialSoftmax,重装解决一切问题。
按照提示安装到conda环境里: 安好后验证一下: 至于nvcc -c的使用: 注意 conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch这里的cudatoolkit和NVIDIA的不是同一个东西,两者都需要安装。 2. 安装cuda参考官方安装教程以及这个博客的第三步和第四步,下载win对应版本的Installer,全部安装后验证PASS。 注意:安装包2.5G,从安装解压到结束至少需要13G的空间,最终安装好的大小为6G左右 3. 安装cuDNN参考官方安装教程,找到对应的版本下载后解压。 4. 验证pytorch和cudatoolkit版本严格对应 参考文章: cuda与显卡驱动版本对照 全流程安装步骤 |
CopyRight 2018-2019 实验室设备网 版权所有 |