再不入坑就晚了,从零学pytorch,一步一步环境搭建

2021-09-08 15:10:35 浏览数 (1)

最近没怎么写游戏了,一直在写python,是因为我对深度学习感兴趣,想学习一下,同时也觉得AI是未来,所以去学习了一段时间。

目的

AI 和游戏的结合是 强化学习,强化学习是深度学习的一个分支,之前也写过一点深度学习,所以这次的学习路线就是由机器学习到深度学习,最后到强化学习,不得不说一条路线对于我来说其实有点难的,深度学习各种公式,各种概念,各种框架,需要时间的积累,所以不是一下就能学会的,在接下来的一段时间还是由浅入深,慢慢的像强化学习逼近,希望能有所产出。

心理准备

鉴于我的本科基础,所以我的文章对公式的解读可能不多,只是从我的理解层面去学习,所以如果你看我写的学习记录,不用担心你数学不好,不用担心公式看不懂,因为我也只是理解皮毛,只要理解公式背后的含义和公式在做什么,我们能在使用的过程中知道自己做什么就可以了。

说实在话那些公式看懂了之后原理也很简单,不用害怕。

IDE的选择

今天是我写的AI第一篇,所以先搞一下环境,在市面上的文章都是用的Jupyter Notebook,但是问了公司做AI的大佬,说是这东西用来做实验,做教程还行,但是对于真正的开发是不合适的,所以我不会选择这种。我选择的编辑器是pycharm。

AI框架的选择

现在最流行的AI学习框架大家应该都知道,一个是TensorFlow 一个是pytorch。

TensorFlow 来自谷歌,pytorch 来自Facebook,两个都是出自名门,各自有各自的优势。

TensorFlow在生产中用的比较多,pytorch用在科研比较多,但是我们公司似乎用的pytorch。

我选择的框架是pytorch,原因是在公司可以找到大佬指教,另外一个原因就是之前写强化学习的时候因为公司网络的限制,当时安装的是pytorch,有一定的经验,没有经验的你也不用担心。

安装环境

1、python的安装

python的安装我选择的是python3.8 ,你可以根据你自己的环境选择,安装教程也不赘述了,一路next ,闪电带火花,没有任何难度

2、pycharm的安装

pycharm的安装就不赘述了。网上的教程一堆一堆的,一路next 加上破解就可以了,网上都有教程,相信没有难度,如果真的遇到困难可以留言和我交流,或者加我微信,公众号的菜单就可以获取我的微信

3、pytorch的安装

因为我写学习记录,也因为我笔记本的缘故,所以不支持gpu训练,我以后换了台式机可能到时会再补上。

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

根据自己的环境选择,生成安装命令,大概率你和我一样是window,不支持GPU,所以和我一样选择就好

在控制台执行 :

代码语言:javascript复制
pip3 install torch torchvision torchaudio

由三种方式执行:

第一种:如果你在安装的时候python加入了环境变量,可以直接 开始按钮,输入cmd

第二种:在pip3.exe 的目录下输入命令

同样打开cmd 命令,输入 cd python 安装目录,进入到python的目录,例如 :

代码语言:javascript复制
cd  C:py38Scripts

或者 直接打开 pip3.exe 的目录,按住shfit 键,然后右键:

然后输入 安装命令

第三种,在pycharm 中输入选择 terminal

等待安装完成就可以了

4、测试环境

测试环境是否正常,在pycharm中创建一个py文件

代码语言:javascript复制
#!/usr/bin/env python
# encoding: utf-8
"""
#Author: 香菜
@time: 2021/8/26 0026 下午 9:59
"""
import  torch
if __name__ == '__main__':
    x = torch.rand(5, 3)
    print(x)

运行不报错就可以了,输出结果如下:

代码语言:javascript复制
tensor([[0.2694, 0.3946, 0.3710],
        [0.0166, 0.1028, 0.3075],
        [0.1616, 0.8462, 0.2754],
        [0.5104, 0.0296, 0.3117],
        [0.9684, 0.9381, 0.1651]])

5、总结

环境的安装虽然很简单,但是对于想要学习的你来说也不容易,毕竟万事开头难,遇到问题很正常,下篇我们写个入门,并且介绍一些概念。准备好一起学习了吗?

0 人点赞