1.首先是安装pytorch
起初,我直接用conda install pytorch命令安装,后来出现的一系列错误让我明白我还是too young too simple,后续说明。
2.安装完pytorch后,就可以在文件所在目录运行python train.py conf/train.json命令,运行结果显示Torch not compiled with CUDA enabled
3.百度一堆,大概理解为未安装cuda,于是用命令conda install cudatoolkit安装cuda,运行后还是显示相同错误。
4.后来在博主https://bbs.csdn.net/topics/392551610的回答中得知是因为安装的pytorch不带cuda,需要安装带cuda的pytorch,于是在博主提供的官网上下载相应版本的pytorch,我用的是conda install pytorch torchvision cudatoolkit=9.0 -c pytorch。其中有三个包需要下载或者更新,最大的是pytorch,400多兆,中途多次由于网络原因中断,导致前面下的全都白搭。所以最好在网络环境好的时候下载。
5.终于经历好几个小时的下载过程后安装好了带有cuda的pytorch,结果又出现了新的错误RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM,目前还在找原因,尚未解决。
2019.7.31更新
经过把anaconda卸载,把所有后来装的东西都卸载,重新在pytorch官网安装符合自己系统配置的带有cuda的pytorch后,终于可以运行了~因此猜测,后来的那个错误还是因为安装的pytorch与系统不兼容的原因。