硬件信息:
系统:win10专业版
CPU:i5-9400f
内存:16GB
显卡:GTX1060
一、打开Anaconda Prompt
1、conda create -n pytorch python=3.7.0:创建名为pytorch的虚拟环境,并为该环境安装python=3.7.0
2、activate pytorch:激活名为pytorch的环境
二、确定硬件支持的CUDA版本
NVIDIA控制面板-帮助-系统信息-组件
可知:GTX1060支持CUDA10版本
三、确定pytorch版本,torchvision版本
1、进入pytorch官网:https://pytorch.org/get-started/locally/
2、确定对应的pytorch和torchvision版本:Package先选择Pip是为了查看pytorch和torchvi