cuda:10.0
环境:pytorch:1.2.0;torchvision:0.4.0
一、下载VS2017
我最开始用的之前安装的VS2019,会出现没有适合的VS版本的提示,下载2017后解决。
二、Anaconda—方便环境管理,可在一个电脑上安装多个环境,不同环境放置不同框架,使用conda create -n创建新环境
anaconda官网下载安装即可,需要配置环境:
三、cuda、cudnn下载和安装
版本是torch1.2.0,对应cuda10.0,cudnn是7.4.15,在官网下载安装即可。
安装完成后在对应的目录下将对应的cudnn进行解压
四、配置pytorch
1、使用conda进行环境创建:
conda create -n pytorch python=3.6<