ubuntu18.04下安裝GPU版pytorch

1、安裝軟件

cuda_10.0.130.1_linux.run
cudnn-10.0-linux-x64-v7.6.4.38.tgz
torch-1.2.0-cp36-cp36m-win_amd64.whl
torchvision-0.4.0-cp36-cp36m-win_amd64.whl

切忌一點:
linux下顯卡要單獨安裝,不要用cuda自帶的顯卡驅動。
windows下可以不用單獨安裝顯卡驅動,用cuda自帶的顯卡驅動就可以。

參考:
ubuntu16.04安裝cuda9.0+cudnn7.3.1+anaconda3+tensorflow1.12
windows10安裝GPU版pytorch步驟(全程離線)
真實機下 ubuntu 18.04 安裝GPU +CUDA+cuDNN 以及其版本選擇

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章