小白入门深度学习 | 第四篇：配置PyTorch环境

Posted 2021-09-14 K同学啊

文章目录

• • 简单的方法：
  • 快捷的方法
  • 检查是否安装成功

简单的方法：

进入 【官网】，通过官方提供的方法进行安装。这个方法的好处是简单

快捷的方法

上面的方法属于pip 在线安装，比较方便，但是安装速度会慢很多。如果希望更快，建议先将安装包文件下载到本地，然后在本地进行安装。 具体方法如下：

第一步：下载安装包文件（.whl文件）

1. 按照上面的方法获取命令A pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html。
2. WIN键 + R 输入 cmd 打开命令行，并运行命令A

3. 在https://download.pytorch.org/whl/torch_stable.html页面中找到上面下载的文件（即安装包文件） cu111/torch-1.9.0%2Bcu111-cp36-cp36m-win_amd64.whl

第二步：直接下载.whl文件

第三步：使用pip进行本地安装

检查是否安装成功

运行下面两份代码，第一份代码运行成功说明你的PyTorch安装成功，第二份代码运行成功并返回True说明你GPU版本的PyTorch安装成功。

import torch
x = torch.rand(5, 3)
print(x)

import torch
torch.cuda.is_available()