0%

PyTorch 学习日志 1

前言

最近的课程有一些 NLP 的内容,为了卑微地跟上课程的进度只好从头安装 PyTorch 以及一些相关的组件,准备入门一些基本操作。

正文

首先找到 PYTorch 的官网,然后参考链接 https://pytorch.org/get-started/locally/

恰好我的环境是之前买的游戏本,用的英伟达显卡,所以可以使用 CUDA 加速计算。(卑微还不太懂 CUDA,之后再做了解吧) 然后就先去安装 CUDA 了。

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

搜到 CUDA 的下载链接后,按照自己的环境选择合适的参数,然后下载安装。中间的高级选项都全部默认了。

image-20200328161706122

上图是即将安装完成前的截图。值得一提的是安装过程大概是会影响显卡驱动……屏幕是真的会一闪一闪一会儿,然后之前的一些配置可能也会清空,比如我之前开的护眼模式,在安装完成后自动取消了,感觉屏幕的颜色很不适应……不过关闭开启护眼模式2次后就好了。

安装完毕后,重新回到 PyTorch 的官网,选择好平台情况,会自动生成合适的命令安装 PyTorch 。

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

别问我为什么用 pip 不用 conda…… 之前因为 conda 出过几次错…… 现在能不用 conda 就尽可能不用了。(坐等以后来打脸)

输入了命令以后就看到了这两行输出……然后是漫长的等待……

1
2
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Looking in links: https://download.pytorch.org/whl/torch_stable.html

几分钟之后……有了新的输出……796 MB?卑微C盘没地方了,该花钱买更大的固态了……

全部完成好了之后,官网有一些方便的测试代码便于测试安装是否成功。

1
2
3
4
5
6
7
8
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

# 检查 CUDA
import torch
torch.cuda.is_available()

开始教程

首先按照官网的指引,打开他们的 60 min Bliz 教程。

https://pytorch.org/tutorials/beginner/blitz/tensor_tutorial.html#sphx-glr-beginner-blitz-tensor-tutorial-py

提示需要读者有 Numpy 基础…… 先忍住继续看。

看完了第一个教程……不知道该写些什么……主要是一些关于 tensor 的操作。