老师需要跑NeuroSim的代码,记录配置环境步骤。

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

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐