pytorch、cuda、python版本對齊的示例分析

發(fā)布時(shí)間：2021-08-03 11:37:48 來源：億速云閱讀：183 作者：小新欄目：開發(fā)技術(shù)

這篇文章主要介紹了pytorch、cuda、python版本對齊的示例分析，具有一定借鑒價(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也能跑，但是一開始可能會(huì)遇到各種各樣的問題，本人cuda版本為10.0，安裝cuda的命令為：

conda install cudatoolkit=10.0

注：GPU顯卡驅(qū)動(dòng)一般沒什么問題，所以盡量不要?jiǎng)觕udnn的版本。

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“pytorch、cuda、python版本對齊的示例分析”這篇文章對大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識等著你來學(xué)習(xí)!

向AI問一下細(xì)節(jié)

pytorch、cuda、python版本對齊的示例分析