下載安裝?CUDA Toolkit
下載地址,這個(gè)工具的下載安裝比較簡單,但是需要注意的是版本問題,因?yàn)闄C(jī)器學(xué)習(xí)開發(fā)庫的支持的cuda 版本不同,所以安裝的同時(shí)注意選擇對應(yīng)的版本避免安裝錯(cuò)誤.下載完成后,默認(rèn)安裝即可.
下載安裝?CUDA Toolkit
下載安裝 nvidia cudnn?下載與CUDA對應(yīng)的CUDNN壓縮包,解壓縮后注意要將 bin 文件夾添加到系統(tǒng) path 中,另外同樣需要注意的是版本問題,各機(jī)器學(xué)習(xí)開發(fā)庫對應(yīng)的cudnn 不盡相同,安裝時(shí)注意避免錯(cuò)誤.