如何在win10環(huán)境中安裝pytorch-gpu-創(chuàng)新互聯(lián)

如何在win10環(huán)境中安裝pytorch-gpu？相信很多沒(méi)有經(jīng)驗(yàn)的人對(duì)此束手無(wú)策，為此本文總結(jié)了問(wèn)題出現(xiàn)的原因和解決方法，通過(guò)這篇文章希望你能解決這個(gè)問(wèn)題。

創(chuàng)新互聯(lián)成立于2013年，是專業(yè)互聯(lián)網(wǎng)技術(shù)服務(wù)公司，擁有項(xiàng)目成都做網(wǎng)站、成都網(wǎng)站建設(shè)網(wǎng)站策劃，項(xiàng)目實(shí)施與項(xiàng)目整合能力。我們以讓每一個(gè)夢(mèng)想脫穎而出為使命，1280元汕城做網(wǎng)站,已為上家服務(wù),為汕城各地企業(yè)和個(gè)人服務(wù),聯(lián)系電話:18982081108

1.Cuda的下載安裝及配置

        首先我們要確定本機(jī)是否有獨(dú)立顯卡。在計(jì)算機(jī)-管理-設(shè)備管理器-顯示適配器中，查看是否有獨(dú)立顯卡。

            可以看到本機(jī)有一個(gè)集成顯卡和獨(dú)立顯卡NVIDIA GetForce GTX 1050。

            接下來(lái)，測(cè)試本機(jī)獨(dú)立顯卡是否支持CUDA的安裝，點(diǎn)擊此處查詢顯卡是否在列表中。

          從上圖中，可以看到我本機(jī)的獨(dú)立顯卡是支持CUDA安裝的，計(jì)算力是6.1。

          那么，接下來(lái)就是選擇CUDA進(jìn)行安裝即可，點(diǎn)擊此處

        這里我選了CUDA Toolkit10.0的版本，至于選擇哪個(gè)版本，個(gè)人認(rèn)為應(yīng)該沒(méi)多大差別，一般就是看這個(gè)版本是否要求GPU的計(jì)算能力是多少以上。大約是2.1G。我用windows的IDM下載工具下載的，速度蠻快的。

       接著就是安裝過(guò)程，雙擊打開(kāi)顯示臨時(shí)解壓目錄，不需要改變，默認(rèn)即可。

         接下來(lái)，進(jìn)入NVIDIA安裝過(guò)程，在這安裝過(guò)程中，我一開(kāi)始直接選擇的精簡(jiǎn)安裝，但由于VS的原因，導(dǎo)致無(wú)法正常安裝，于是我換成了自定義的安裝方式，并將VS勾給去掉，便可以正常安裝了，至于CUDA的安裝目錄，大家默認(rèn)安裝在C盤(pán)即可。

        安裝完成之后，便是配置環(huán)境變量。環(huán)境變量配置如下圖所示。

       測(cè)試CUDA是否正常安裝?？梢钥吹綔y(cè)試成功，接著進(jìn)行cudnn的下載。

2.CUDNN的下載及配置

        點(diǎn)擊此處，選擇下載download cudnn，但這里需要你注冊(cè)一個(gè)賬號(hào)，然后進(jìn)行問(wèn)卷之后才可以進(jìn)行下載頁(yè)面，反正一步步操作即可。

       然后因?yàn)槲疑弦徊紺UDA的版本是10.0,而CUDNN的版本要跟CUDA版本一致，所以選擇第二個(gè)下載即可。

         下載之后，解壓縮，將CUDNN壓縮包里面的bin、clude、lib文件直接復(fù)制到CUDA的安裝目錄下，直接覆蓋安裝即可。

3.pytorch-gpu的安裝

        直接來(lái)到Pytorch的官網(wǎng)，此處，因?yàn)楦鶕?jù)自己的配置進(jìn)行選擇，我試過(guò)用conda安裝，但是conda安裝老是不成功，也替換過(guò)清華鏡像源，但速度實(shí)在是龜速，半天沒(méi)動(dòng)靜。雖然pip下載也蠻慢，但至少可以安裝成功。這里我對(duì)pip的安裝方式，稍作修改，讓其直接從清華鏡像源下載。然后慢慢等待安裝成功即可。

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

       然后命令行驗(yàn)證pytorch是否正常安裝成功，這里可以正常打印出版本號(hào)，安裝沒(méi)問(wèn)題。

import torch
print(torch.__version__)

 

         但接下來(lái)我嘗試讓其輸出是否可以正常調(diào)用CUDA，卻輸出False.

print(torch.cuda.is_available())

        經(jīng)過(guò)一段時(shí)間的交流，找到問(wèn)題所在。因?yàn)槲冶緳C(jī)是集成顯卡和獨(dú)立顯卡并存，可能是因?yàn)槟J(rèn)調(diào)用集成顯卡而造成的問(wèn)題，那么接下來(lái)就是將獨(dú)立顯卡設(shè)為默認(rèn)即可。

        但我因?yàn)槎啻螄L試的結(jié)果，成功將我本機(jī)的NAVIDIA控制面板給卸載掉了，無(wú)法正常打開(kāi)，所以我還需要重新下載NVIDIA的控制面板，來(lái)到官網(wǎng)此處，選擇對(duì)應(yīng)的版本下載安裝即可。

         然后右鍵打開(kāi)NIVIDIA控制面板，選擇管理3D設(shè)置，全局設(shè)置中選擇高性能NVIDIA處理器即可。

        然后在命令行下重新輸入

import torch
print(torch.cuda.is_available())

       即可返回True，若還返回False的話 ，記得重啟一下電腦，遇到問(wèn)題重啟一下電腦，看是否可以正常顯示。

看完上述內(nèi)容，你們掌握如何在win10環(huán)境中安裝pytorch-gpu的方法了嗎？如果還想學(xué)到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝各位的閱讀！

網(wǎng)頁(yè)標(biāo)題：如何在win10環(huán)境中安裝pytorch-gpu-創(chuàng)新互聯(lián)
網(wǎng)站URL：http://bm7419.com/article16/hcgdg.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供企業(yè)網(wǎng)站制作、定制開(kāi)發(fā)、營(yíng)銷(xiāo)型網(wǎng)站建設(shè)、微信公眾號(hào)、全網(wǎng)營(yíng)銷(xiāo)推廣、網(wǎng)站收錄

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源： 創(chuàng)新互聯(lián)