Windows下安装pytorch gpu版本

pytorch gpu版本安装

自己安装后的一个总结

1.下载安装anaconda
2.查看cuda版本号 ，终端中命令 nvidia-smi查看
3.下载cuda 命令nvcc --version 查看是否安装成功
一般不要修改默认安装的位置，即使修改也未必能安装到指定位置
4.下载cudnn
5.cudnn拷贝到cudn的 bin,lib,include
注意：不是替换文件夹，而是将文件放入对应的文件夹中
6. 配置环境变量
将bin所在的目录添加到环境变量 PATH 中，“此电脑”→‘属性–“高级系统设置”→“环境变量”→“系统变量”→“path”→“编辑”→“新建”加入该路径即可。
加入四个变量—网上找
7.pytorch安装
pytorch的主流安装方式有两种，分别是conda和pip。
conda: 依托强大的Anaconda集成开发环境（IDE）而存在，上手调试很方便，适合新手。
pip: 轻量级，纯净，适合喜欢简约风格的人，不过调试不方便。(这里指的是纯python环境下的pip)
查看conda下的环境 conda info --envs
（1） 创建一个独立的运行环境，方便开发过程中遇到的多版本并存 conda create -n pytorch_gpu pip python=3.6
pytorch_gpu 为环境名字，可自定义，python3.6为该环境中的版本，与安装Anaconda时默认送的python互不干扰，
如果想使用其他版本的python，直接更改“=”后的版本号即可。
（2） 进入我们创建好的环境中 conda activate pytorch_gpu
（3） 安装pytorch，去官网找到相应的安装命令
（4）验证是否成功，返回true则成功。

     import torch
     torch.cuda.is_available()