在使用深度學(xué)習(xí)模型訓(xùn)練的過(guò)程中，工具的準(zhǔn)備也算是一個(gè)良好的開(kāi)端吧。熟話說(shuō)完事開(kāi)頭難，磨刀不誤砍柴工，先把前期的問(wè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也能跑，但是一開(kāi)始可能會(huì)遇到各種各樣的問(wèn)題，本人cuda版本為10.0，安裝cuda的命令為：
|
1
|
conda install cudatoolkit=10.0 |
注：GPU顯卡驅(qū)動(dòng)一般沒(méi)什么問(wèn)題，所以盡量不要?jiǎng)觕udnn的版本。
以上這篇淺談pytorch、cuda、python的版本對(duì)齊問(wèn)題就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持服務(wù)器之家。
原文鏈接：https://blog.csdn.net/weixin_38664232/article/details/95449106
