win10 英偉達顯卡MAX150 安裝GPU版pytorch(一)選擇合適的CUDA、CUDNN和torch

本人深度學習框架小白一枚,在安裝使用深度學習框架過程中遇到了不少問題,幸好有師兄的幫助,才得以成功安裝。同時也深感對於新入門的小夥伴,在安裝pytorch過程中的不易,特此將經驗分享一下,如果覺得不錯,請點個贊吧。
本人電腦的基本配置是:2018年4月份買的電腦,英偉達顯卡MAX150,固態硬盤 256G,Intel i5-8250u, 使用的是python3.7 ,IDLE是pycharm
安裝的pytorch 的版本是 :torch-1.0.1-cp37-cp37m-win_amd64
安裝的內容:CUDA CUDNN 和torch
具體的安裝步驟如下:
1、你的電腦必須有英偉達(NVIDIA)的顯卡,這是必備的基本條件。第一步,有的博客說你需要到nvidia的官網查看你的顯卡是否支持的CUDA,個人認爲是沒有必要的。你需要做的是在桌面右鍵打開控制面板,在這裏插入圖片描述
點擊左下角的“系統信息”→再點擊“組件”在這裏插入圖片描述
這樣就查到了你的顯卡是否支持CUDA以及適合的版本是哪種。之後就可以去NVIDIA的官網https://developer.nvidia.com/cuda-toolkit-archive下載對應版本。(這是我的顯卡支持的版本,但是我不建議使用太新的版本。所以我下載的是9.2版的。)在這裏插入圖片描述
其次就是下載對應版本的CUDNNhttps://developer.nvidia.com/cudnn。注意,兩者的版本一定要對應!在下載CUDNN時需要先進行註冊,比較簡單一分鐘就完事了。之後就下載相應的版本就行了。在這裏插入圖片描述
在這裏插入圖片描述
然後到pytorch的官網https://pytorch.org/下載對應版本的torch在這裏插入圖片描述
這裏需要注意選擇你的電腦系統,安裝包類型、你使用的編譯語言以及CUDA的版本。(雖然我下載的CUDA是9.2的,但是這裏選擇9.0的torch也是支持的)
這樣我們需要安裝的軟件包已經下載完畢了。

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