win10+cuda10.2+python3.8+pytorch环境配置

一、安装CUDA

前阵给电脑刷机，准备重新学一下深度学习，先从比较简单的pytorch框架入手，利用电脑1060显卡加速训练，记录配置环境需要注意的问题。

首先需要确定电脑是支持cuda的，查询本机显卡是否支持CUDA的安装，是否存在列表中，
网址:https://developer.nvidia.com/cuda-legacy-gpus.

如果有下载你想要的cuda版本，我下载的是10.2版。点击进入网址https://developer.nvidia.com/cuda-toolkit-archive.选择想要下载的版本，进行下载。

 下载文件为exe可执行文件，点击执行。

 然后点击ok安装，安装选项选自定义，勾去VS。

 

安装完成后检查是否安装成功。

 二、安装pytorch

 首先确定要安装的版本，进入官网 https://pytorch.org/get-started/locally/

这里选择最新版本，win10，pip安装，cuda10.2。因为文件过大，选择将包下载下来，再进行本地安装 。

复制图中下载链接，在浏览器中粘贴打开，因为本python版本为3.8，故选择cu102，cp38，win，amd64位点击下载。一定要选择正确版本，否则会安装失败。

 下载完成后利用pip进行本地安装，打开cmd窗口，输入代码

pip3 install 文件名,例：

pip3 install D:\python\Scripts\torch-1.9.0+cu102-cp38-cp38-win_amd64.whl

三、检查

安装成功后在cmd界面输入python进入python环境，输入以下代码确认是否安装正确。

import torch

print(torch.__version__)

print(torch.cuda.is_available())

 显示True说明安装成功。

四、参考网址

Pytorch + Win10系统 + pip安装+ CUDA9.1版本（安装CUDA10.2版本）_CSDN1683的博客-CSDN博客

Start Locally | PyTorch