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

5439905 mindspore 1578992805

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.]

 

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小宋是呢

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值