Win10环境下安装PyTorch GPU版【CUDA8.0+CUDNN8.0】

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Andone_hsx/article/details/87946183

承接上篇博文win10环境下《深度学习之CUDA与CUDNN详细安装教程》https://blog.csdn.net/Andone_hsx/article/details/87942968

本文主要详细介绍PyTorch的安装过程和各种问题解决

PyTorch官网:https://pytorch.org/

在安装PyTorch之前,我们需要安装好Anaconda:https://www.anaconda.com/distribution/

双击下载并安装,注:选择默认模式,一直next安装!

安装好Anaconda后,打开PyTorch官网,选择相关对应的PyTorch版本

进入cmd, 复制并运行相应版本PyTorch下载安装执行语句

(上图红框内的语句:conda install pytorch torchvision cudatoolkit=8.0 -c pytorch)

等待安装......

安装过程中会出现,下载慢,下载不成功,即使设置了清华镜像,还是下到一半断掉

CondaError: CondaHTTPError: HTTP 000 CONNECTION FAILED

可通过选择设置清华镜像+翻墙来解决此问题。

设置清华镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes
在Anaconda中按行执行,如果第一次设置,不会出现任何提示!

在翻墙的情况下,重新执行conda install pytorch torchvision cudatoolkit=8.0 -c pytorch即可!

测试PyTorch是否安装成功:

进入python IDE下

分别执行:

1、import torch

2、torch.__version__

3、torch.cuda.is_available()

注意:torch.cuda.is_available()是个函数,输出True,则表示cuda、cudnn、PyTorch均安装成功!

建议在使用Anaconda时创建虚拟环境,在虚拟环境下安装PyTorch!

相关博文:

Anaconda创建虚拟环境

【Win10环境】深度学习之CUDA与CUDNN详细安装教程

猜你喜欢

转载自blog.csdn.net/Andone_hsx/article/details/87946183