计算神经信息内容 · 入门 运行单元测试： python unit_test.py 运行主脚本： python main.py 使用Jupyter笔记本make_plots.ipynb生成图。 环境细节 该代码在以下位置运行： 火炬1.5.0 使用Docker容器nvcr.io/nvidia/pytorch:20.03-py3 在NVIDIA Titan RTX GPU上，驱动程序版本为440.82，CUDA版本为：10.2 引文 如果您认为此代码有用，请随时引用： misc