It seems pytorch0.2.0 are not supported under cuda9.0

baowenbo / MEMC-Net

MEMC-Net: Motion Estimation and Motion Compensation Driven Neural Network for Video Interpolation and Enhancement

MIT License

170 stars 22 forks source link

It seems pytorch0.2.0 are not supported under cuda9.0 #1

Closed pz0910 closed 5 years ago

pz0910 commented 5 years ago

Are you sure about the environment requirement: pytorch0.2.0_4+cuda9.0+cudnn7.0? I failed to install pytorch0.2.0 with cuda9.0. Can you help me out? And I found a misspelling in README: './install.sh' should be replaced with './install.bash'

baowenbo commented 5 years ago

So sorry, i made a mistake in the README. Pytorch0.2 should be cuda8.0 with cudnn5.0. I also provide an environment.yml file for reference. Thank you~

baowenbo commented 5 years ago

Sure we have to use './install.bash'.