PyTorch Docker映像 Ubuntu + PyTorch + CUDA(可选) 要求 为了使用此映像,您必须安装Docker Engine。 上了设置Docker Engine的说明。 CUDA要求 如果您具有兼容CUDA的NVIDIA图形卡,则可以使用启用CUDA的PyTorch映像版本来启用硬件加速。 我仅在Ubuntu Linux中对此进行了测试。 首先,请确保您安装了适当的NVIDIA驱动程序。 在Ubuntu上,我发现确保安装正确版本的驱动程序的最简单方法是,通过安装至少与要使用的映像一样新的CUDA版本。 例如,如果打算使用cuda-10.1图像,则设置CUDA 10