pytorch-gpu安裝的經(jīng)驗(yàn)與教訓(xùn)

更新時(shí)間：2023年01月13日 08:23:56 作者：故y

本文主要介紹了pytorch-gpu安裝的經(jīng)驗(yàn)與教訓(xùn)，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

首先說明

本文并不是安裝教程，網(wǎng)上有很多，這里只是自己遇到的一些問題

我是以前安裝的tensorflow-gpu的，但是發(fā)現(xiàn)現(xiàn)在的學(xué)術(shù)論文大部分都是用pytorch復(fù)現(xiàn)的，因此才去安裝的pytorch-gpu

查看自己安裝的CUDA

nvcc -V

這里我提供一個(gè)安裝tensorflow時(shí)所用的CUDA對(duì)應(yīng)表

安裝cuDNN時(shí)版本一定要對(duì)應(yīng)正確

安裝完CUDA時(shí)要改一下環(huán)境變量

只用添加這個(gè)就好，其他三個(gè)CUDA會(huì)自動(dòng)為你添加

我的是cuda-11.2

但是官網(wǎng)沒有配套的，直接就挑自己喜歡的下就行Start Locally | PyTorch

這里強(qiáng)調(diào)：千萬不要相信在其他網(wǎng)站上的下載命令。有可能你下載的就是cpu版本

安裝的時(shí)候可能會(huì)遇到

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

不要擔(dān)心，不用管它，耐心等一等，就會(huì)出現(xiàn)（y/n）選擇y就可以了

這里要看一眼你的pytorch是不是cuda版本的，不是要重新下載

測(cè)試安裝是否成功

import torch
print(torch.__version__)  #注意是雙下劃線
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

當(dāng)然你這些都做完了，還是出現(xiàn)了False

失敗后的操作

打開你的anaconda

將這里關(guān)于pytorch的信息全部刪除，再按照前面的流程走一遍，相信你自己一定會(huì)出現(xiàn)

到此這篇關(guān)于pytorch-gpu安裝的經(jīng)驗(yàn)與教訓(xùn) 的文章就介紹到這了,更多相關(guān)pytorch-gpu安裝內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

arcgis使用Python腳本進(jìn)行批量截圖功能實(shí)現(xiàn)
最近公司數(shù)據(jù)部那邊有個(gè)需求，需要結(jié)合矢量數(shù)據(jù)和影像數(shù)據(jù)，進(jìn)行批量截圖，并且截圖中只能有一個(gè)圖斑，還要添加上相應(yīng)的水印，這篇文章主要介紹了arcgis使用Python腳本進(jìn)行批量截圖,需要的朋友可以參考下
2023-01-01
python 定時(shí)器每天就執(zhí)行一次的實(shí)現(xiàn)代碼
這篇文章主要介紹了python 定時(shí)器每天就執(zhí)行一次的實(shí)現(xiàn)代碼，代碼簡(jiǎn)單易懂非常不錯(cuò)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
2019-08-08
簡(jiǎn)單了解python反射機(jī)制的一些知識(shí)
這篇文章主要介紹了簡(jiǎn)單了解python反射機(jī)制的一些知識(shí)，反射機(jī)制就是在運(yùn)行時(shí)，動(dòng)態(tài)的確定對(duì)象的類型，并可以通過字符串調(diào)用對(duì)象屬性、方法、導(dǎo)入模塊，是一種基于字符串的事件驅(qū)動(dòng)。,需要的朋友可以參考下
2019-07-07
python中的list字符串元素排序
這篇文章主要介紹了python中的list字符串元素排序方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2022-11-11
python并發(fā)2之使用asyncio處理并發(fā)
本篇文章主要介紹了python并發(fā)2之使用asyncio處理并發(fā)，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2017-12-12
python如何對(duì)圖片或文件的操作
在日常編程中,我們經(jīng)常會(huì)遇到文件處理的需求,如base64與圖片的相互轉(zhuǎn)換、圖片與像素點(diǎn)的操作、本地文件與二進(jìn)制的互相轉(zhuǎn)換、計(jì)算文件的md5以及下載網(wǎng)絡(luò)文件等,這些操作對(duì)于處理多媒體數(shù)據(jù)、實(shí)現(xiàn)數(shù)據(jù)的持久化存儲(chǔ)、保證數(shù)據(jù)的完整性和安全性等方面都至關(guān)重要
2024-09-09
Python編程使用NLTK進(jìn)行自然語言處理詳解
這篇文章主要介紹了Python編程使用NLTK進(jìn)行自然語言處理詳解，涉及了nltk和開發(fā)環(huán)境的簡(jiǎn)單介紹，以及SentencesSegment，SentencesSegment等內(nèi)容，具有一定參考價(jià)值，需要的朋友可以了解下。
2017-11-11
Diango + uwsgi + nginx項(xiàng)目部署的全過程（可外網(wǎng)訪問）
這篇文章主要給大家介紹了關(guān)于Diango + uwsgi + nginx項(xiàng)目部署的全過程（可外網(wǎng)訪問），文中通過示例代碼將部署的過程介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。
2018-04-04
Python編程pygal繪圖實(shí)例之XY線
這篇文章主要介紹了Python編程pygal繪圖實(shí)例之XY線，具有一定借鑒價(jià)值，需要的朋友可以參考下。
2017-12-12
Django 連接sql server數(shù)據(jù)庫(kù)的方法
這篇文章主要介紹了Django 連接sql server數(shù)據(jù)庫(kù)的方法,小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2018-06-06