PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置

PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置

  • 前期准备
    • 1、电脑必须支持NVIDIA独立显卡并且已经安装独显驱动
    • 2、设置首选图形处理器
    • 3、查看独显支持的Cuda版本
  • 安装Cuda9.0
    • 1、官网下载Cuda9.0
    • 2、安装Cuda9.0
    • 3、验证cuda安装
  • 安装 cuDNN v7.3.1
    • 1、官网下载
    • 2、安装cuDNN v7.3.1
  • 安装Anaconda3-5.2.0
    • 1、安装Anaconda3-5.2.0
    • 2、验证anaconda安装
  • 利用Anaconda Navigator配置多环境Anaconda3-5.2.0
    • 1、启动Anaconda Navigator
    • 2、新增运行环境
    • 3、激活运行环境(PyToch)
    • 4、设置清华镜像
  • 安装 PyTorch 0.4.1 GPU版本
    • 1、官网查询安装命令
    • 2、在PyToch环境下运行
    • 3、验证 PyTorch-GPU 安装
  • PyCharm 的环境配置

前期准备

1、电脑必须支持NVIDIA独立显卡并且已经安装独显驱动

8G以下就不用考虑安装了),如下图所示
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第1张图片

2、设置首选图形处理器

在NVIDIA 控制面板中 -> 选择管理 3D 设置 -> 全局设置 -> 首选图形处理器中选择高性能 NVIDIA 处理器,然后点击应用
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第2张图片

3、查看独显支持的Cuda版本

非常重要!不支持PyTorch会运行不了!
点击帮助 -> 系统信息 -> 组建 查看
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第3张图片
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第4张图片

安装Cuda9.0

1、官网下载Cuda9.0

官网:https://developer.nvidia.com/cuda-90-download-archive
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第5张图片

2、安装Cuda9.0

运行cuda_9.0.176_win10.exe
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第6张图片
选择自定义安装
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第7张图片
在选择安装项时一般不安装GeForce Experience,CUDA是核心组件必须勾上,剩下两个选项的当前版本如果比新版本低的话也可以勾选上。接着点击下一步就开始安装了。
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第8张图片

3、验证cuda安装

win+r -> cmd 运行nvcc -V查看
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第9张图片
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第10张图片

安装 cuDNN v7.3.1

1、官网下载

官网:https://developer.nvidia.com/cudnn
选择 Download cuDNN -> Download cuDNN v7.3.1 (Sept 28, 2018), for CUDA 9.0 -> cuDNN v7.3.1 Library for Windows 10
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第11张图片
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第12张图片

2、安装cuDNN v7.3.1

解压下载cudnn-9.0-windows10-x64-v7.3.1.20.zip
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第13张图片
将这三个文件拷贝到CUDA9.0的安装路径根文件夹下
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第14张图片

安装Anaconda3-5.2.0

1、安装Anaconda3-5.2.0

运行Anaconda3-5.2.0-Windows-x86_64.exe
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第15张图片
勾选第一项自动添加环境变量
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第16张图片
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第17张图片

2、验证anaconda安装

win+r -> cmd 运行 conda -V 查看
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第18张图片
在这里插入图片描述

利用Anaconda Navigator配置多环境Anaconda3-5.2.0

1、启动Anaconda Navigator

PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第19张图片

2、新增运行环境

Environments -> Create 输入 环境名称(PyTorch)选择python版本(3.7)
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第20张图片

3、激活运行环境(PyToch)

运行 Anaconda Prompt -> 输入 activate PyTorch 则可将环境从base切换到PyToch
在这里插入图片描述
在这里插入图片描述

4、设置清华镜像

清华镜像官网:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
输入以下命令

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

conda config --set show_channel_urls yes

安装 PyTorch 0.4.1 GPU版本

1、官网查询安装命令

官网:https://pytorch.org/
点击 Get Started
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第21张图片
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第22张图片

2、在PyToch环境下运行

conda install pytorch -c pytorch

pip3 install torchvision

# for no installed pip3
pip install torchvision

PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第23张图片
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第24张图片

3、验证 PyTorch-GPU 安装

运行以下命令

python

import torch

torch.cuda.is_available()

在这里插入图片描述
torch.cuda.is_available() 若返回值为True ,恭喜你安装成功~
若返回值为False,安装失败,恭喜你可以更换配套版本重新安装啦~
至此 Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本已经安装完成~
以下是 PyCharm 的环境配置

PyCharm 的环境配置

Settings -> Project Interpreter
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第25张图片
点击右上角小齿轮 选择 Add
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第26张图片
在这里插入图片描述
选择System Interpreter -> 选择右上角的 …
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第27张图片
选择 Anaconda 安装目录下的 envs -> PyTorch -> python.exe
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第28张图片
点击ok
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第29张图片
出现 pytorch 则配置完成
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第30张图片
验证配置 新建 GPUTest.py 文件 并运行

import torch

print(torch.__version__)

cuda = torch.cuda.is_available()
print(cuda)

a = torch.Tensor((3, 5))
print(a)
print(a.cuda())

配置成功
PyTorch学习一 —— Win10环境下安装 NVIDIA Cuda9.0 + 多环境Anaconda3-5.2.0 + PyTorch 0.4.1 GPU版本 + PyCharm环境配置_第31张图片

你可能感兴趣的:(PyTorch)