torch에서 제공하는 mixed precision을 써보려면 이사이트에서 이야기한 것처럼 pytorch 1.6.0 버전 이상이 필요한 것 같습니다.
그래서 최신 버전(1.8.0)으로 업그레이드
=> segmentation models library랑 torch 버전이 안 맞는 듯(정확한 파악은 아닙니다)
=> 실패!
중간 버전(1.6.0)으로 업그레이드
=> NVIDIA dirver 안 맞음
=> 실패!
AssertionError:
The NVIDIA driver on your system is too old (found version 10010).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.
time 모듈로 1epoch학습시간 보니깐 60초 가량 걸리길래 그냥 사용 안 해도 되겠다 싶어서 사용 안 하려구요.
혹시 SMP라이브러리랑해서 잘 되신 분은 나중에 알려주세요
torch에서 제공하는 mixed precision을 써보려면 이사이트에서 이야기한 것처럼 pytorch 1.6.0 버전 이상이 필요한 것 같습니다. 그래서 최신 버전(1.8.0)으로 업그레이드 => segmentation models library랑 torch 버전이 안 맞는 듯(정확한 파악은 아닙니다) => 실패!
중간 버전(1.6.0)으로 업그레이드 => NVIDIA dirver 안 맞음 => 실패!
time 모듈로 1epoch학습시간 보니깐 60초 가량 걸리길래 그냥 사용 안 해도 되겠다 싶어서 사용 안 하려구요. 혹시 SMP라이브러리랑해서 잘 되신 분은 나중에 알려주세요