docker上的“ CUDA驱动程序版本不足以支持CUDA运行时版本”

时间:2019-07-24 18:52:08

标签: docker cuda docker-compose pytorch

我的系统具有Ubuntu 18.04,Cuda 10.1,最新版本的CuDNN,1080Ti GPU和32 GB Ram。

我想运行以下代码:https://github.com/longcw/MOTDT

首先,我安装了deepo的最新版本,这是一个包含许多深度学习库的docker映像。它写在代码的自述文件中,首先应执行“ make.sh”。我试过了,但是它给出了一些错误。当我在存储库页面上查看“问题”时,建议使用Cuda 9.0和Pytorch 0.4来解决该问题。我无法运行代码,因为我已经安装了该映像的最新版本,并且包括Cuda 10和PyTorch的最新版本,而代码则需要Cuda 9和PyTorch <= 0.4。

第二,我删除了Deepo的最新版本,并安装了包含Cuda 9和PyTorch(版本> 0.4)的Deepo版本“ all-py36-cu90”。在其他运行代码中,我通过以下命令在docker映像上安装了PyTorch 0.4:

pip install torch==0.4 -f https://download.pytorch.org/whl/cu90/stable 

因为它被写入代码的自述文件中,所以我执行了“ make.sh”,成功了。但是,当我运行“ eval_mot.py”时,出现此错误:

  

CUDA驱动程序版本不足于CUDA运行时版本

我不知道如何解决这个问题。另外,我不想直接在系统上运行代码,而是想在docker上运行。

0 个答案:

没有答案