『AI实践学』Ubuntu安装MindSpore-gpu1.0踩坑之旅

小宋是呢

发布于 2020-10-29 11:15:33

9040

发布于 2020-10-29 11:15:33

文章被收录于专栏：深度应用

0.引子

测试环境

ubuntu18.4
cuda10.2
使用conda创建cudatoolkit=10.1

测试代码：

import numpy as np
from mindspore import Tensor
from mindspore.ops import functional as F
import mindspore.context as context

context.set_context(device_target="GPU")
x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = Tensor(np.ones([1,3,3,4]).astype(np.float32))
print(F.tensor_add(x, y))

1.问题解决

1.1 ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory

解决方法：

1.首先通过：conda list 确认是否安装 cudatoolkit=10.1

如果没有的话，执行： conda install cudatoolkit=10.1 cudnn=7

2.增加 LD_LIBRARY_PATH：

先通过： which python 确定当前conda环境路径，笔者的输出：/root/miniconda3/envs/tf2c/bin/python

在增加LD_LIBRARY_PATH：export LD_LIBRARY_PATH="/root/miniconda3/envs/tf2c/lib"

此时第一个问题就解决了

1.2 OSError: MindSpore version 1.0.0 and cuda version 10.2 does not match, reference to the match info on: https://www.mindspore.cn/install

详细错误信息为：

Traceback (most recent call last):
  File "test_img.py", line 5, in <module>
    from mindspore import Tensor
  File "/root/miniconda3/envs/tf2c/lib/python3.7/site-packages/mindspore/__init__.py", line 17, in <module>
    from ._version_check import check_version_and_env_config
  File "/root/miniconda3/envs/tf2c/lib/python3.7/site-packages/mindspore/_version_check.py", line 256, in <module>
    check_version_and_env_config()
  File "/root/miniconda3/envs/tf2c/lib/python3.7/site-packages/mindspore/_version_check.py", line 250, in check_version_and_env_config
    env_checker.check_version()
  File "/root/miniconda3/envs/tf2c/lib/python3.7/site-packages/mindspore/_version_check.py", line 95, in check_version
    raise EnvironmentError(f"MindSpore version {__version__} and cuda version {v_str} does not match, "
OSError: MindSpore version 1.0.0 and cuda version 10.2 does not match, reference to the match info on: https://www.mindspore.cn/install

解决方法：修改 /root/miniconda3/envs/tf2c/lib/python3.7/site-packages/mindspore/_version_check.py 文件

注释一下几行：

        """
        if v_str not in self.version:
            raise EnvironmentError(f"MindSpore version {__version__} and cuda version {v_str} does not match, "
            "reference to the match info on: https://www.mindspore.cn/install")
        """

问题解决。执行代码，输出：

[[[[2. 2. 2. 2.]
   [2. 2. 2. 2.]
   [2. 2. 2. 2.]]

  [[2. 2. 2. 2.]
   [2. 2. 2. 2.]
   [2. 2. 2. 2.]]

  [[2. 2. 2. 2.]

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020/10/23 ，如有侵权请联系 cloudcommunity@tencent.com 删除

python

深度学习

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

python

深度学习

登录后参与评论

0 条评论

热度