win10+GeForce 940mx安装cuda
生活随笔
收集整理的這篇文章主要介紹了
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。
注意
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與顯卡驅動版本對照
全流程安裝步驟
總結
以上是生活随笔為你收集整理的win10+GeForce 940mx安装cuda的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 《微信朋友圈,这么玩才赚钱》读书笔记-刘
- 下一篇: http上传文件 pda java_通过