windows下用GPU跑python

1、查看CUDA版本
控制面板小图标找英伟达 nvidia控制面板
或者cmd

nvidia-smi

注意下面这个版本号不是想查看的

nvcc --version

没有需安装，版本号以控制面板显示结果为准

2、安装pycharm
安装pycharm以及解释器
解释器

3、安装anaconda
链接： https://www.anaconda.com/download/

4、conda虚拟环境配置
cmd进入 Anaconda3\Script

conda create -p address\name python=3.6

3.6可以换成别的版本，address为虚拟盘文件路径，name是虚拟环境地址

或者

conda create -n name python=3.6

此时默认路径在anaconda\envs

5、pytorch安装
默认stable、操作系统、语言、CUDA版本（nvidia控制器为准）选择，选pip

复制匹配语句到cmd

6、active
查看确认

conda-env list

active

conda activate  address\name

7、查看GPU内存
cmd

nvidia-smi