淺談pytorch、cuda、python的版本對齊問題
在使用深度學習模型訓練的過程中,工具的準備也算是一個良好的開端吧。熟話說完事開頭難,磨刀不誤砍柴工,先把前期的問題搞通了,能為后期節(jié)省不少精力。
以pytorch工具為例:
pytorch版本為1.0.1,自帶python版本為3.6.2
服務器上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顯卡驅動一般沒什么問題,所以盡量不要動cudnn的版本。
以上這篇淺談pytorch、cuda、python的版本對齊問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
詳解python中DRF框架的數(shù)據(jù)校驗方式
這篇文章主要為大家詳細介紹了python中DRF框架的數(shù)據(jù)校驗方式,文中的示例代碼講解詳細,具有一定的學習價值,感興趣的小伙伴可以跟隨小編一起了解一下2023-10-10Python+Jmeter實現(xiàn)自動化性能壓測的流程步驟
性能測試是一個全棧工程師/架構師必會的技能之一,只有學會性能測試,才能根據(jù)得到的測試報告進行分析,找到系統(tǒng)性能的瓶頸所在,而這也是優(yōu)化架構設計中重要的依據(jù),本文給大家介紹了Python+Jmeter實現(xiàn)自動化性能壓測的流程步驟,需要的朋友可以參考下2024-05-05Pandas DataFrame操作數(shù)據(jù)增刪查改
我們在用 pandas 處理數(shù)據(jù)的時候,經(jīng)常會遇到用其中一列數(shù)據(jù)替換另一列數(shù)據(jù)的場景。這一類的需求估計很多人都遇到,當然還有其它更復雜的。解決這類需求的辦法有很多,這里我們來推薦幾個,這篇文章主要介紹了Pandas DataFrame操作數(shù)據(jù)的增刪查改2022-10-10基于Python中isfile函數(shù)和isdir函數(shù)使用詳解
今天小編就為大家分享一篇基于Python中isfile函數(shù)和isdir函數(shù)使用詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11