PyTorch相关资源

PyTorch相关资源

  1. 官网

平台不同,pytorch的安装方式也不同。

  1. 官方文档

这是英文版的文档,update-to-date。中文版的版本较老,且因为是机翻,错误很多。比如把Tip提示翻译成小费我也是醉了,翻完都不检查一下的吗,技术性的东西容不得错误,还是看官方的文档吧。

  1. 本地Jupyter Docker环境
docker run -it --name jupyter -p 1808:8888 -e JUPYTER_ENABLE_LAB=yes -e GRANT_SUDO=yes --user root  -v /local/path/jupyter:/home/jovyan/ -d jupyter/minimal-notebook

Jupyter Lab比老的Jupyter Notebook更好用,功能更强大。 同时以root权限登录。挂载一个本地目录到/home/jovyan,方便本地编辑。

需要安装一些依赖才可以运行官方demo

# pytorch torchvision torchaudio
pip3 install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# matplotlab
pip3 install matplotlib

# ipywidgets,以支持Progress Bar。装完之后需要重启相关Kernel。
pip3 install ipywidgets

# pandas
pip3 install pandas
  1. 开源生态

其中不乏一些好的项目。

  1. 移动端

先在PC端测试好模型,再导到移动端,方便快捷。

  1. PyTorch Lightning

一个对PyTorch二次封装,类似于Keras的框架。

Leave a Comment

豫ICP备19001387号-1