在安裝pytorch時,怎麼判斷要安裝gpu版本的還是不安裝?
首先默認是NVIDIA的顯卡
pytorch要使用顯卡的話需要安裝NVIDIA的驅動,然後根據NVIDIA官網下載相應的:
cuda(CUDA Toolkit 11.1 Update 1 Downloads)
cudnn(NVIDIA cuDNN)需要註冊
然後根據installation guide進行安裝。
接下來才安裝pytorch,根據cuda的版本選擇對應的pytorch版本進行安裝
首先默認是NVIDIA的顯卡
pytorch要使用顯卡的話需要安裝NVIDIA的驅動,然後根據NVIDIA官網下載相應的:
cuda(CUDA Toolkit 11.1 Update 1 Downloads)
cudnn(NVIDIA cuDNN)需要註冊
然後根據installation guide進行安裝。
接下來才安裝pytorch,根據cuda的版本選擇對應的pytorch版本進行安裝