這篇文章主要介紹了pytorch、cuda、python版本對齊的示例分析，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。
以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版本對齊的示例分析”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
