深度学习(二):Pytorch框架安装讲解

1、前言 Pytorch分为CPU版和GPU版,工做中通常会选择GPU进行训练。这里咱们主要介绍如何安装Pytorch的GPU版。由于版本要求比较严格,我这里搭配了一套可用的版本,可当作参考。win10 + 1070ti + Anaconda + python3.6 + CUDA10.1 + cuDNN8.0.2 + pytorch1.5.0python 2、win10 操做系统我这里选择win1
相关文章
相关标签/搜索