My old torch environment which lets CSRNet training run as fast as usual.

environment_old.yml

name: torchold
channels:
  - albumentations
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - albumentations=0.4.0=py36_0
  - geos=3.7.2=he1b5a44_2
  - giflib=5.1.4=0
  - graphite2=1.3.11=0
  - imageio=2.6.1=py36_0
  - imgaug=0.3.0=py_0
  - jupyter_contrib_core=0.3.3=py_2
  - jupyter_contrib_nbextensions=0.5.0=py36_1000
  - jupyter_highlight_selected_word=0.2.0=py36_1000
  - jupyter_latex_envs=1.4.4=py36_1000
  - jupyter_nbextensions_configurator=0.4.0=py36_1000
  - jupyterlab=0.35.4=py36_0
  - jupyterlab_server=0.2.0=py_0
  - libiconv
  - libwebp=0.5.2=7
  - libxslt=1.1.32=h88dbc4e_2
  - lxml=4.2.5=py36hc9114bc_0
  - mkl_fft=1.0.10=py36_0
  - mkl_random=1.0.2=py36_0
  - openblas=0.2.20=8
  - pixman=0.34.0=2
  - pywavelets=1.1.1=py36hc1659b7_0
  - scikit-image=0.14.2=py36hf484d3e_0
  - shapely=1.6.4=py36hec07ddf_1006
  - tbb=2019.9=hc9558a2_0
  - tbb4py=2019.9=py36hc9558a2_0
  - x264=20131218=0
  - _libgcc_mutex=0.1=main
  - astroid=2.3.1=py36_0
  - attrs=19.1.0=py36_1
  - autopep8=1.4.4=py_0
  - backcall=0.1.0=py36_0
  - blas=1.0=mkl
  - bleach=3.1.0=py36_0
  - bokeh=0.12.16=py36_0
  - bzip2=1.0.6=h9a117a8_4
  - ca-certificates=2019.10.16=0
  - cairo=1.14.12=h8948797_3
  - certifi=2019.9.11=py36_0
  - cffi=1.11.5=py36h9745a5d_0
  - click=6.7=py36h5253387_0
  - cloudpickle=0.5.3=py36_0
  - cudatoolkit=9.0=h13b8566_0
  - cudnn=7.1.2=cuda9.0_0
  - cycler=0.10.0=py36h93f1223_0
  - cytoolz=0.9.0.1=py36h14c3975_0
  - dask=0.17.4=py36_0
  - dask-core=0.17.4=py36_0
  - dbus=1.13.2=h714fa37_1
  - decorator=4.4.0=py36_1
  - defusedxml=0.6.0=py_0
  - distributed=1.21.8=py36_0
  - entrypoints=0.3=py36_0
  - expat=2.2.5=he0dffb1_0
  - ffmpeg=4.0=hcdf2ecd_0
  - fontconfig=2.13.0=h9420a91_0
  - freeglut=3.0.0=hf484d3e_5
  - freetype=2.9.1=h8a8886c_1
  - glib=2.56.2=hd408876_0
  - gmp=6.1.2=h6c8ec71_1
  - gst-plugins-base=1.14.0=hbbd80ab_1
  - gstreamer=1.14.0=hb453b48_1
  - h5py=2.8.0=py36ha1f6525_0
  - harfbuzz=1.8.8=hffaf4a1_0
  - hdf5=1.10.2=hba1933b_1
  - heapdict=1.0.0=py36_2
  - html5lib=1.0.1=py36h2f9c1c0_0
  - icu=58.2=h9c2bf20_1
  - intel-openmp=2018.0.0=8
  - ipykernel=5.1.2=py36h39e3cac_0
  - ipython=7.8.0=py36h39e3cac_0
  - ipython_genutils=0.2.0=py36_0
  - ipywidgets=7.5.1=py_0
  - isort=4.3.21=py36_0
  - jasper=2.0.14=h07fcdf6_1
  - jedi=0.15.1=py36_0
  - jinja2=2.10.1=py36_0
  - jpeg=9b=h024ee3a_2
  - jsonschema=3.0.2=py36_0
  - jupyter=1.0.0=py36_7
  - jupyter_client=5.3.3=py36_1
  - jupyter_console=6.0.0=py36_0
  - jupyter_core=4.5.0=py_0
  - kiwisolver=1.0.1=py36h764f252_0
  - lazy-object-proxy=1.4.2=py36h7b6447c_0
  - libedit=3.1.20181209=hc058e9b_0
  - libffi=3.2.1=hd88cf55_4
  - libgcc=7.2.0=h69d50b8_2
  - libgcc-ng=9.1.0=hdf63c60_0
  - libgfortran=3.0.0=1
  - libgfortran-ng=7.2.0=hdf63c60_3
  - libglu=9.0.0=hf484d3e_1
  - libopencv=3.4.2=hb342d67_1
  - libopus=1.2.1=hb9ed12e_0
  - libpng=1.6.37=hbc83047_0
  - libprotobuf=3.5.2=h6f1eeef_0
  - libsodium=1.0.16=h1bed415_0
  - libstdcxx-ng=9.1.0=hdf63c60_0
  - libtiff=4.0.9=he85c1e1_1
  - libuuid=1.0.3=h1bed415_2
  - libvpx=1.7.0=h439df22_0
  - libxcb=1.13=h1bed415_1
  - libxml2=2.9.8=hf84eae3_0
  - locket=0.2.0=py36h787c0ad_1
  - markupsafe=1.1.1=py36h7b6447c_0
  - matplotlib=2.2.2=py36hb69df0a_2
  - mccabe=0.6.1=py36_1
  - mistune=0.8.4=py36h7b6447c_0
  - mkl=2018.0.3=1
  - msgpack-python=0.5.6=py36h6bb024c_0
  - nbconvert=5.6.0=py36_1
  - nbformat=4.4.0=py36_0
  - nccl=1.3.5=cuda9.0_0
  - ncurses=6.1=hf484d3e_0
  - networkx=2.1=py36_0
  - ninja=1.8.2=py36h6bb024c_1
  - notebook=6.0.1=py36_0
  - numpy=1.15.4=py36h1d66e8a_0
  - numpy-base=1.15.4=py36h81de0dd_0
  - olefile=0.45.1=py36_0
  - opencv=3.4.2=py36h6fd60c2_1
  - openssl=1.1.1d=h7b6447c_3
  - packaging=17.1=py36_0
  - pandas=0.23.0=py36h637b7d7_0
  - pandoc=1.19.2.1=hea2e7c5_1
  - pandocfilters=1.4.2=py36_1
  - parso=0.5.1=py_0
  - partd=0.3.8=py36h36fd896_0
  - pcre=8.42=h439df22_0
  - pexpect=4.7.0=py36_0
  - pickleshare=0.7.5=py36_0
  - pillow=5.4.1=py36h34e0f95_0
  - pip=19.2.3=py36_0
  - prometheus_client=0.7.1=py_0
  - prompt_toolkit=2.0.9=py36_0
  - psutil=5.4.5=py36h14c3975_0
  - ptyprocess=0.6.0=py36_0
  - py-opencv=3.4.2=py36hb342d67_1
  - pycodestyle=2.5.0=py36_0
  - pycparser=2.18=py36hf9f622e_1
  - pygments=2.4.2=py_0
  - pylint=2.4.2=py36_0
  - pyparsing=2.2.0=py36hee85983_1
  - pyqt=5.6.0=py36h22d08a2_6
  - pyrsistent=0.15.4=py36h7b6447c_0
  - python=3.6.9=h265db76_0
  - python-dateutil=2.8.0=py36_0
  - pytorch=0.4.0=py36hdf912b8_0
  - pytz=2018.4=py36_0
  - pyyaml=3.12=py36hafb9ca4_1
  - pyzmq=17.1.2=py36h14c3975_0
  - qt=5.6.3=h8bf5577_3
  - qtconsole=4.5.5=py_0
  - readline=7.0=h7b6447c_5
  - scikit-learn=0.19.1=py36h7aa7ec6_0
  - scipy=1.1.0=py36hfa4b5c9_1
  - send2trash=1.5.0=py36_0
  - setuptools=41.2.0=py36_0
  - simplegeneric=0.8.1=py36_2
  - sip=4.18.1=py36hf484d3e_2
  - six=1.12.0=py36_0
  - sortedcontainers=1.5.10=py36_0
  - sqlite=3.29.0=h7b6447c_0
  - tblib=1.3.2=py36h34cf8b6_0
  - terminado=0.8.2=py36_0
  - testpath=0.4.2=py36_0
  - tk=8.6.8=hbc83047_0
  - toolz=0.9.0=py36_0
  - tornado=6.0.3=py36h7b6447c_0
  - traitlets=4.3.2=py36_0
  - typed-ast=1.4.0=py36h7b6447c_0
  - wcwidth=0.1.7=py36_0
  - webencodings=0.5.1=py36_1
  - wheel=0.33.6=py36_0
  - widgetsnbextension=3.5.1=py36_0
  - wrapt=1.11.2=py36h7b6447c_0
  - xz=5.2.4=h14c3975_4
  - yaml=0.1.7=had09818_2
  - zeromq=4.2.5=h439df22_0
  - zict=0.1.3=py36h3a3bf81_0
  - zlib=1.2.11=h7b6447c_3
  - cuda90=1.0=h6433d27_0
  - torchvision=0.2.1=py36_1