win10+GeForce 940mx安装cuda 您所在的位置:网站首页 940m驱动 win10+GeForce 940mx安装cuda

win10+GeForce 940mx安装cuda

2023-09-17 11:12| 来源: 网络整理| 查看: 265

起因是在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. 验证

5. GPU不起作用 可能遇到的问题:

pytorch和cudatoolkit版本严格对应

参考文章:

cuda与显卡驱动版本对照

全流程安装步骤



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有