谷歌JAX深度学习从零开始学
上QQ阅读APP看书,第一时间看更新

1.2.4 使用PyCharm和JAX

下面使用PyCharm进行程序设计,在进行程序设计之前,需要在PyCharm中加载JAX编译环境,编译步骤说明如下。

 在桌面新建一个空文件夹,命名为“JaxDemo”。启动PyCharm并打开刚才新建的空文件夹“JaxDemo”,结果如图1.15所示。

图1.15 启动PyCharm

此时并没有使用WSL中的Python进行解释,请继续进行下一步工作。

 在Windows环境下启动PyCharm,依次选择“Setting|Project|ProjectInterpreter|Python Interpreter”选项,然后单击图1.16所示窗口右侧的操作按钮,准备更改Python的解释器。

 单击“Add”按钮,加载WSL中的Python解释器,如图1.17所示。

 之后在弹出的界面左侧的类别中选择“WSL”,右侧路径改成如图1.18所示的内容。

图1.16 更改Python解释器

注意

这里的默认路径地址是python,而读者在WSL中使用的Python版本是python3。

图1.17 添加Python解释器

图1.18 加载WSL中的Python3解释器

 单击“OK”按钮,PyCharm开始加载文件,界面如图1.19所示。

图1.19 编译器配置完成

等编译结束后就可以进入PyCharm的程序代码阶段,至此配置完成。