首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Anaconda+vscode+pytorch环境搭建

好像是真的新年祝福

大好猴~

胡博士在这里携QQ祝大家狗年大吉(懒了这么久,才写一篇,还不认真排版……其实只是简单用了md)……

假装我是前言

最近微软和Anaconda达成了合作,新版的conda整合了vscode这个特别好用的编辑器……具体可以点这里。博士一直受不了pycharm每次index时的风扇狂转(有时index也不及时,很诡异),也受不了Atom的连接缓慢(我就是懒得XX)和node.js引起的风扇狂转(是是是,我风扇破),所以一直用vscode写python,虽然功能上欠缺一些,好像也不能远程调试,但是一直比较轻量舒服,而且Monokai主题也挺好看的,看起来像Sublime

开始说正题

废话不多说,开始我们的配置,我应该写得比较通俗,毕竟本人比较笨。

需要准备的环境

Anaconda(官网下载)

vscode(我装的时候还是从微软下的,没有用conda管理)

flake8与yapf(PiPy安装即可,我没拼错吧?)

pytorch(写这篇文章时pytorch的版本为0.3,facebook官方没有提供win版编译好的,好在有各路大神,一会儿再说这个)

安装Anaconda和vscode

这一步没什么难度,Windows版的Anaconda和vscode直接双击安装程序和下一步就OK了,新版的anaconda安装时的默认路径应该没有空格了,如果有,你就自己改一下路径,不要出现空格,不然有得你麻烦的。

由于tensorflow rdkit pytorch等等这些都已经支持了py36,所以我都直接下最新带python3.6的conda,python2的官方支持没几年就停了吧,而且现在新出的很多包都提供python3,如果你不是接手很旧的项目,又只想学一种,那就3吧

anaconda在安装结束后会问你是不是要加到path路径里,如果这是你机器上唯一的python环境,那就加呗,加了之后还可以用conda创建和管理其他版本的环境,多好。

pip install

教你个装X的做法打开win10的命令行,win+x组合键,然后按一下A

打开命令行

输入“pip install flake8”

输入“pip install yapf”(你可以一起装,我就是让你看到,这里装了2)

vscode中的配置是以json格式存储,并且用户配置会覆盖系统配置,用起来非常灵活,有很多选项,你尽可以随便改~

安装pytorch

Linux和Mac直接按照官方的文档安装,很简单,和tf一样,torch从lua移植过来之后,也优先支持了linux,从github上看到的消息是,0.4(也就是下一个版本)开始支持windows,所以我们先用别人编译好的~当然用起来一样的~~

最初我用的peterj123的conda包,但是在code, pycharm中都会出一些问题,搞了半天也没搞明白是哪里的问题,然后又找了另一个,就是下边说的这个。出处网址是这个 https://zhuanlan.zhihu.com/p/31747695 关注的话可以follow一下他的github(顺便也follow一下胡博士的呗~ github id: jecinghu)。

打开命令行

输入“conda install numpy mkl pyyaml cffi”

输入“pip install torch-0.3.0b0.591e73e-cp36-cp36m-win_amd64.whl”(我有个940mx的残卡,可以cuda加速,cuda和cudnn配置这里就不写了,如果你没有,可以找找cpu版的pytorch包,如果你有卡不会配置,那你去屎吧!)

输入“pip install torchvision”(顺序不要乱啊,不然你可能装不上这个)

随便打开python或者ipython交互,输入可以输入这些测试一下

没有错误应该就可以了。

配置code

总算到这步了,vscode令人惊奇的就是,打开一个几百兆的文件的速度远远快于sublime和atom(当然还有自带的记事本),让我这种半吊子不能不爱

首先准备这样的文件结构

在tasks.json中写入这些

如果你用的python不在你的path之中,commad的值就把绝对路径(全部路径,整条路径,整个廊)写上吧

可以把.vscode(注意这个文件夹名字前面有个“.”,千万不要落下)备份一下,有新的python项目直接往项目的根文件夹里扔一个这个文件夹,然后code就可以直接运行看结果了……

可以去我的github找我弄好的.vscode中的文件拉下来也行

https://github.com/jecinghu/drugbank/tree/master/.vscode

在code里装上需要的扩展,按Ctrl+Shift+X组合键打开扩展窗口,然后搜索Anaconda Extension Pack,目测装上这个之后python扩展和yaml支持,没有的话你就自己装一个python扩展吧,开发者是Microsoft的那个

配置用户设置

在code中打开项目文件夹

可以在vscode中打开project文件夹(注意是打开文件夹,而不是test.py文件),也可以在Windows资源管理器(Win+E组合键)的project文件夹界面,右键,点选“Open with code”

在code的资源管理器选择程序运行入口

在code的资源管理器(Ctrl+Shift+E组合键)中选择想要运行的脚本(这里是test.py),然后按Ctrl+Shift+B组合键,再选好什么python之类的(这里我记不住了,我这里直接就是好了的)就可以了,我们这里简单运行一个回归,可以看到左边的资源管理器有.vscode文件夹,下方正常输出~~~ 完美~!

我这里用matplotlib做了个简单的可视化,在服务器上我一般会用visdom配合pytorch(Tensorflow(不要在意大小写,我写得着急)有TensorBoard),你选一个适合你的就好~

就酱,Bye!

  • 发表于:
  • 原文链接http://kuaibao.qq.com/s/20180220G06LWM00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券