淺談pytorch、cuda、python的版本對齊問題
在使用深度學(xué)習(xí)模型訓(xùn)練的過程中,工具的準(zhǔn)備也算是一個良好的開端吧。熟話說完事開頭難,磨刀不誤砍柴工,先把前期的問題搞通了,能為后期節(jié)省不少精力。
以pytorch工具為例:
pytorch版本為1.0.1,自帶python版本為3.6.2
服務(wù)器上GPU的CUDA_VERSION=9000
注意:由于GPU上的CUDA_VERSION為9000,所以至少要安裝cuda版本>=9.0,雖然cuda=7.0~8.0也能跑,但是一開始可能會遇到各種各樣的問題,本人cuda版本為10.0,安裝cuda的命令為:
conda install cudatoolkit=10.0
注:GPU顯卡驅(qū)動一般沒什么問題,所以盡量不要動cudnn的版本。
以上這篇淺談pytorch、cuda、python的版本對齊問題就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
解決python中畫圖時x,y軸名稱出現(xiàn)中文亂碼的問題
今天小編就為大家分享一篇解決python中畫圖時x,y軸名稱出現(xiàn)中文亂碼的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01使用Python正則表達(dá)式操作文本數(shù)據(jù)的方法
這篇文章主要介紹了使用Python正則表達(dá)式操作文本數(shù)據(jù)的方法,小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2019-05-05Python連接PostgreSQL數(shù)據(jù)庫并查詢數(shù)據(jù)的詳細(xì)指南
在現(xiàn)代軟件開發(fā)中,數(shù)據(jù)庫是存儲和檢索數(shù)據(jù)的核心組件,PostgreSQ是一個功能強(qiáng)大的開源對象關(guān)系數(shù)據(jù)庫系統(tǒng),它以其穩(wěn)定性、強(qiáng)大的功能和靈活性而聞名,Python作為一種流行的編程語言,與PostgreSQL的結(jié)合使用非常廣泛,本文介紹了Python連接PostgreSQL數(shù)據(jù)庫并查詢數(shù)據(jù)2024-12-12