安裝GPU版PyTorch

這段時間準備用pytorch來跑深度學習,之前在anaconda上直接下載的GPU版似乎不能直接使用,問了師兄,原來不僅僅裝一個GPU版的pytorch就可以,還要安裝對應顯卡的一些東西,這過程費了很多時間,所以想把它總結在這裏,希望想安裝的人少走一點彎路。 1 在NVIDIA官網下載顯卡對應的CUDA驅動 下載地址:https://developer.nvidia.com/cuda-downlo
相關文章
相關標籤/搜索