安装完毕后显示成功,但是使用pycharm用ssh连接后不能跑官网给的示例代码
收藏回复举报
安装完毕后显示成功,但是使用pycharm用ssh连接后不能跑官网给的示例代码
t('forum.solved') 已解决
发表于2024-03-16 09:38:25
0 查看
使用
nvidia-smi

得到结果NVIDIA-SMI 535.154.05   Driver Version: 535.154.05   CUDA Version: 12.2。

在自己的容器内安装的是image.png

在服务器的ubuntu容器里安装成功,但是使用pycharm并ssh连接时运行失败。失败结果如下:

import numpy as np
import mindspore as ms
import mindspore.ops as ops

if __name__ == '__main__':
    ms.set_context(device_target="GPU")
    x = ms.Tensor(np.ones([1,3,3,4]).astype(np.float32))
    y = ms.Tensor(np.ones([1,3,3,4]).astype(np.float32))
    print(ops.add(x, y))

image.png

请问有没有解决办法?

我要发帖子