记录一下我遇到的问题。
我在安装cupy运行的时候出现这个问题;
我运行的环境是Ubuntu,由于同时装了cuda9.0和cuda10.0,导致cupy安装出现了问题。

首先
import cupy as cp
print(cp.show_config())

结果
在这里插入图片描述如果这里CUDA同时有9.0和10.0可能就会出现CUDA_ERROR_INVALID…的问题。

我当时是CUDA driver version 是10.0 其他是9.0,查了一下区别 https://blog.csdn.net/ljp1919/article/details/102640512。

最后解决办法就是,把9.0换到了10.0。

Logo

一站式 AI 云服务平台

更多推荐