eisen-ai
commented
4 years ago Hello,
It is necessary to have pytorch 1.6 in order to use AMP.
You can still use other pytorch features. In order to do so, please use a (slightly) older version of eisen (one that has an older version of torch as a requirement)..
Consider updating your CUDA or simply updating the driver and running docker!
On 27. Sep 2020, at 05:52, Sniper notifications@github.com wrote:
Hi，My PC have the CUDA Version is 10.0. so i check online find that : pytorch version must be 1.2. but if i change the torch version to 1.2, the Eisen will be prompt ModuleNotFoundError: No module named 'torch.cuda.amp', i find that reason is Eisen and Pytorch version not match!!!
so, how can i Configuration these ones version？
My configuration： ubuntu 18.04LST CUDA 10.0 torch 1.2 Eisen 0.1.10
Tk u
— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/eisen-ai/eisen-core/issues/38, or unsubscribe https://github.com/notifications/unsubscribe-auth/AOERTVAB2HTSB2XF2L2PPIDSH2ZJBANCNFSM4R3JU6VQ.
Hi，My PC have the CUDA Version is 10.0. so i check online find that : pytorch version must be 1.2. but if i change the torch version to 1.2, the Eisen will be prompt
ModuleNotFoundError: No module named 'torch.cuda.amp', i find that reason is Eisen and Pytorch version not match!!!so, how can i Configuration these ones version？
My configuration： ubuntu 18.04LST CUDA 10.0 torch 1.2 Eisen 0.1.10
Tk u