深度學習(二):Pytorch框架安裝講解

1、前言 Pytorch分爲CPU版和GPU版,工做中通常會選擇GPU進行訓練。這裏咱們主要介紹如何安裝Pytorch的GPU版。由於版本要求比較嚴格,我這裏搭配了一套可用的版本,可當作參考。win10 + 1070ti + Anaconda + python3.6 + CUDA10.1 + cuDNN8.0.2 + pytorch1.5.0python 2、win10 操做系統我這裏選擇win1
相關文章
相關標籤/搜索