老师需要跑NeuroSim的代码,记录配置环境步骤。
- 拉取源码
https://github.com/neurosim/DNN_NeuroSim_V2.0 - 到Training_pytorch 文件夹下,运行inference.py文件
发现报错:cannot find utee
在utee内加一个空python文件,名为__init__.py即可解决。 - 下载anaconda全家桶,新建虚拟环境
conda create -n pytorch python=3.5
命名为pytorch(随便) - 下载pytorch,注意版本为1.0.0,并下载对应的cuda版本
可以去官网下载 - 运行python train.py