最近发现利用pytorch进行模型训练会有版本不兼容的情况,一直走在配环境的大路上,后来接触了docker,利用docker 沙箱机制可以freestyle 进行版本组合进行训练学习。
这个网址里面包含了不少可用来做训练的cuda环境。
https://hub.docker.com/r/nvidia/cuda/tags?page=1&ordering=last_updated
最近发现利用pytorch进行模型训练会有版本不兼容的情况,一直走在配环境的大路上,后来接触了docker,利用docker 沙箱机制可以freestyle 进行版本组合进行训练学习。
这个网址里面包含了不少可用来做训练的cuda环境。
https://hub.docker.com/r/nvidia/cuda/tags?page=1&ordering=last_updated