Pytorch——Window上Pytorch的安装教程

1 前言

推荐下我自己建的人工智能Python学习群:[809160367]，群里有我整理的一份关于pytorch、python基础，图像处理opencv\自然语言处理、机器学习、数学基础等资源库，想学习人工智能或者转行到高薪资行业的，大学生都非常实用，无任何套路免费提供！还可以扫码加VX领取资料哦！

 

2 环境要求

Windows：

CUDA（先要装好CUDA库）

3 安装步骤

Windows：

step 1：首先进入Pytorch官网，根据电脑的配置信息，获得相应的安装指令，

安装链接：Start Locally | PyTorch

如图所示，

tep 2：安装过程中可能会出现因为网络问题而下载不成功的情况（下载速度很慢，并且在中途会出现下载不成功的情况，which is due to the GFW），下到一半的时候会出现“CondaHTTPError: HTTP 000 CONNECTION FAILED for url”的错误，从而无法继续安装，这时候我们需要使用镜像来进行安装，

这里我们使用的是清华大学的Conda镜像，我们进入清华大学的Conda镜像网站，

网站告诉我们需要修改.condarc文件，我们按照网站中给出的内容进行替换，如下图所示，

 

 4 如何验证安装成功

1、打开 Anaconda Prompt；

2、输入

python

以进入python的编译环境；

3、分别输入以下代码：

import torch

import torchvision

print(torch.cuda.is_available())

这段代码的含义是：引入相关的包，并检测CUDA架构是否可用。

如果CUDA可用，就会得到如下的输出，

 

输出“True”就说明，CUDA可以在Pytorch中使用了，

测试成功啦！

6 问题排除
torch.cuda.is_available()输出False

从直观上来说，就是CUDA在Pytorch框架中无法使用，原因可能有以下几点：

 
显卡驱动版本低，不支持当前版本的CUDA
CUDA版本不正确
显卡驱动版本低，不支持当前版本的CUDA：

在使用 CUDA Toolkit安装CUDA时，在“自定义”选项中， CUDA Toolkit会自动为你匹配最新最合适的驱动版本，如果当前电脑安装的驱动版本低于Toolkit推荐的版本的话，就需要更新最新的显卡驱动。

CUDA版本不正确：

即Pytorch的版本与CUDA版本不对应，这在Pytorch下载的时候就应当进行确认。

这里再次推荐下我自己建的人工智能Python学习群:[809160367]，群里有我整理的一份关于pytorch、python基础，图像处理opencv\自然语言处理、机器学习、数学基础等资源库，想学习人工智能或者转行到高薪资行业的，大学生都非常实用，无任何套路免费提供！还可以扫码加VX领取资料哦！