GTX1050 安装GPU版pytorch流程

• 版本安装情况

Windows10+NVIDIA GTX 1050（笔记本版）+ Driver Version: 471.41 + CUDA 10.1 + python 3.7 + conda 4.10.1 + pytorch 1.7.1

• Anaconda安装

官网进行下载：https://www.anaconda.com/

创建虚拟环境：conda create --name 环境名 python=版本号（我的执行为conda create --name pytorch python=3.7）

查看虚拟环境：conda info --envs

添加清华的镜像源：

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 --set show_channel_urls yes

• CUDA下载

要找好自己显卡对应的CUDA版本，不建议下载最新的CUDA，因为可能与框架存在不匹配的问题。

本次下载的版本为CUDA10.1

接下来安装CUDA，选择自定义安装，我在E盘中建立了文件夹CUDA用于存放自定义安装时的内容（将其中一步中的前两个安装目录放在CUDA1，第三个安装目录放在CUDA2）：

• cuDNN安装

安装版本与CUDA10.1相对应，存在多个与CUDA10.1对应的cuDNN，选择其中一个非最新的，我选择的是cuDNN7.6.5

下载下来的为一个压缩包，将压缩包的内容复制到刚才创立的文件夹CUDA的CUDA1中（注意路径，粘贴到CUDA1中）

• pytorch安装

使用pip的安装方法，进入到虚拟环境pytorch中进行安装（在cmd中使用指令conda activate pytorch），在pytorch中复制指令进行安装：

使用验证程序进行验证，发现安装成功：