win10平台下通过Anaconda安装GPU版PyTorch

本文参考:(8条消息) Ubuntu16.04下安装Cuda、Cudnn、Pytorch gpu版本、torchvision,一篇到位,亲测可行,含泪总结!!!_深度检测的博客-CSDN博客

Anaconda5.2.0安装

下载地址:Index of / (anaconda.com)

1
2
3

修改文件位置

4
5

注意,这里勾选下第一个选项从而添加到环境变量中,免得自己手动添加

6

这样环境变量中的用户变量,已自动添加好

7

开始安装

8

打开Anaconda Prompt

9

创建一个虚拟环境,输入conda create -n pytorch python=3.6

10

新建环境遇到的问题解决:(8条消息) 解决CondaHTTPError:HTTP 000 CONNECTION FAILED for url<https://mirrors.tuna.tsinghua.edu.cn/anaconda***_洲的学习笔记的博客-CSDN博客

channels:
  - defaults
show_channel_urls: true
default_channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

在C盘用户文件夹下找到.condarc文件用记事本打开并修改

11

12

激活pytorch环境,conda activate pytorch

13

虚拟环境所在位置

14

显示所有环境信息,输入conda info --envs

在这里插入图片描述

cuda10.1和cudnn7.6.3安装

使用官网命令安装pytorch时其实不用提前装cuda和cudnn,但为了后续装tensorflow之类的python库,这里先给装上

cuda和cudnn具体指导安装细节:TensorFlow2.1.0安装教程

先查看下自己的显卡驱动,win+r加回车,输入nvidia-smi

在这里插入图片描述

结合下面图选定cuda10.1版本,显卡驱动明显满足要求,同时cuda10.1支持的pytorch版本也较多

在这里插入图片描述

cuda10.1下载地址:CUDA Toolkit 10.1 original Archive | NVIDIA Developer,点击下载,默认地址安装在C盘

在这里插入图片描述

取消勾选两个选项,Visual Studio IntegrationDisplay Driver

在这里插入图片描述

安装位置默认就行

在这里插入图片描述

安装结束打开命令行检查是否安好,输入nvcc -V

在这里插入图片描述

cudnn7.6.3下载地址:cuDNN Archive | NVIDIA Developer,解压后将里面的cuda文件夹重命名为cudnn763,直接复制到cuda安装目录下:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

在这里插入图片描述

点击继续,提供管理员权限

在这里插入图片描述

在这里插入图片描述

并设置好环境变量

在这里插入图片描述

验证安装成功:(8条消息) Windows10检查Cuda和cuDNN是否安装成功?_jhsignal的博客-CSDN博客_windows查看cudnn是否安装成功

在这里插入图片描述

在这里插入图片描述

pytorch1.4.0安装

失败安装经历

这里我参考开头那篇博客中,去清华源将两个离线包手动下载下来安装,因为很多博客或帖子说使用官网的命令并使用清华源,会被conda镜像给坑了,它不是下载的gpu版本,而是cpu版本,于是我就没直接用官网命令安装,选择了离线安装

官网历史版本地址:Previous PyTorch Versions | PyTorch

在这里插入图片描述

pytorch1.4.0和torchvision0.5.0清华源地址:Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

在这里插入图片描述

在这里插入图片描述

将清华源的两个包下载并放在D:\anaconda\pkgs

在这里插入图片描述

在这里插入图片描述

在pytorch虚拟环境中安装两个离线包的指令:

  1. pytorch1.4.0:conda install --offline D:\anaconda\pkgs\pytorch-1.4.0-py3.6_cuda101_cudnn7_0.tar.bz2

  2. torchvision0.5.0:conda install --offline D:\anaconda\pkgs\torchvision-0.5.0-py36_cu101.tar.bz2

期间有个小错误,我直接输入包报错:PathNotFoundError,可是conda install不是去D:\anaconda\pkgs找包的吗,不理解。
另外说下为什么环境变成了PyTorch,别问,问就是踩坑太多,换了多次虚拟环境
在这里插入图片描述

输入两个包的绝对地址,成功安装上

在这里插入图片描述

在这里插入图片描述

检查是否成功,显然失败并报错:from torch._C import * ImportError: DLL load failed:找不到指定模块

在这里插入图片描述

去网上找原因,两篇参考帖子:https://stackoverflow.com/questions/50612899/from-torch-c-import-importerror-dll-load-failed-the-specified-module-could,https://blog.csdn.net/linchunmian/article/details/80520145

于是照第一篇回答,用conda命令下载intel-openmp:Intel Openmp :: Anaconda.org,结果下载速度特别慢,依旧报错

照第二篇回答,直接下载包并配置环境变量,Files :: Anaconda.org,依旧报错

经过我的研究,可以排除intel-openmp的原因,不能解决问题

无奈只得删除环境,输入指令:conda env remove -n your-env-name

在这里插入图片描述

成功安装GPU版本

重新新建pytorch环境,再次在基础环境中输入:conda create -n pytorch python=3.6

这次我先直接试下官网历史版本1.4.0的conda代码,据说会装成cpu版本,看下会不会被conda镜像给坑了

# CUDA 10.1
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

直接一次性成功,给我整不会了都

在这里插入图片描述

好像是因为没走清华源,conda list后channel 为pytorch,事实证明这就是gpu版本的安装指令

在这里插入图片描述

根据个人判断,使用官网的命令时但去掉了结尾的-c pytorch,结果是走清华源,会被conda镜像给坑了,装的都是cpu版本(当然这我没试过)

因此,只需要安装好Anaconda,新建一个虚拟环境,安装pytorch不需要提前安装cudacudnn(当然安装好没什么问题,装tensorflow不是更好),直接按照官网提供的指令进行安装(我是上午装的,速度还挺快的),最后检查下是不是gpu版

# CUDA 10.1
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
import torch
torch.cuda.is_available()

PyCharm安装

pycharm历史版本地址:Other Versions - PyCharm (jetbrains.com)

参考文章:(8条消息) 【PyTorch教程】P2. Python编辑器的选择、安装及配置_我是土堆的博客-CSDN博客_pytorch编译器

在这里插入图片描述

这里修改下地址

在这里插入图片描述

打上两个勾

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述
双击打开pycharm
在这里插入图片描述
使用建好的虚拟环境
在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

ok,大功告成!

猜你喜欢

转载自blog.csdn.net/Star_ID/article/details/123694232