基于Pytorch的深度学习 —

文章目录

硬件环境

安装版本

安装步骤

0、更新显卡驱动

1、查看GPU驱动版本

2、查看GPU支持CUDA版本

3、安装Anaconda

4、安装Python

5、安装CUDA

6、安装CUDNN

7、安装Pytorch

8、验证安装

硬件环境

CPU：Intel Core i7-8750H CPU @2.2GHz 2.21GHz

GPU：NVIDIA GeForce GTX 1050 Ti

RAM：8G

OS：Windows 10 家庭中文版

安装版本

GPU显卡驱动：451.67

Python：3.8.0

Anaconda：Anaconda3-2020.11

CUDA：10.2.89

CUDNN：8.0.5

Pytorch：1.7.1

torchvision：0.6.0

安装步骤

0、更新显卡驱动

任务管理器中查看GPU显卡，本机为GTX 1050Ti，安装驱动版本：451.67；具体安装步骤可查看：安装Nvidia显卡驱动

如果一直安装失败，查看是否之前的版本未删除干净，可以用DDU工具清理，查看：显卡驱动卸载工具DDU

1、查看GPU驱动版本

桌面右键 —> NVIDIA控制面板 —> “帮助”菜单 —> “系统信息” —> “显示”显示框 —> 驱动程序版本

2、查看GPU支持CUDA版本

桌面右键 —> NVIDIA控制面板 —> “帮助”菜单 —> “系统信息” —> “组件”显示框 —> “NVCUDA64.DLL”

3、安装Anaconda

基于Python 3.8；

如果安装使用离线下载包安装Pytorch可不安装Anaconda；

下载地址：https://www.anaconda.com/products/individual

直接安装即可，注意安装路径最好不要有空格；

4、安装Python

下载地址：https://www.python.org/downloads/windows/

安装完成后，打开命令提示符，输入“python”，如果显示版本号，则表示安装成功，如下：

5、安装CUDA

查询适合本机配置的CUDA版本：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

下载地址：CUDA Toolkit 10.2 Download

本次选择安装版本：cuda_10.2.89_441.22_win10.exe

安装步骤查看：安装CUDA

验证是否安装成功：打开命令提示符输入"nvcc -V"，注意nvcc后面有空格，最后V字母要大写，如下图所示：

6、安装CUDNN

下载地址：https://developer.nvidia.com/rdp/cudnn-download

CUDNN版本要同CUDA SDK对应，本次选择安装版本：CUDNN v8.0.5(November 9th, 2020), for CUDA 10.2

安装步骤查看：安装CUDNN

7、安装Pytorch

7.1、先下载文件，再本地安装；

下载地址：https://download.pytorch.org/whl/torch_stable.html

选择需要的版本，包括torch和torchvision，我的版本：

cu102/torch-1.7.0-cp38-cp38-win_amd64.whl
cu102/torchvision-0.6.0-cp38-cp38-win_amd64.whl

注意保存的路径不要有空格，否则会报错：

打开 Anaconda Prompt，输入命令安装Pytorch；

pip install ‪D:\Software\DeepLearning\torch-1.7.1-cp38-cp38-win_amd64.whl

继续在Anaconda Prompt窗口中输入命令安装torxhvision；

pip install D:\Software\DeepLearning\torchvision-0.6.0-cp38-cp38-win_amd64.whl

7.2、使用官网地址下载，速度贼慢；

官网地址：https://pytorch.org/get-started/locally/

选择需要的版本：

复制上述命令：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

打开Anaconda Prompt；输入复制的命令；

输入“y”开始安装；等待安装完成；

8、验证安装

完成安装后，验证是否成功；

打开Anaconda Prompt窗口中，进入python环境，然后输入如下命令，如果没有报错，则表示安装成功；

import torch