请问如何 TinyMS 如何使用 GPU？

[songquanpeng]

我已经配置好了 CUDA 10.1 环境，安装好了 MindSpore GPU 1.2.0 版本。
然后安装 TinyMS 时发现它还是去装了 CPU 版本的 MindSpore：

Collecting mindspore==1.2.0
  Downloading mindspore-1.2.0-cp37-cp37m-manylinux1_x86_64.whl (85.7 MB)

之后：

>>> from tinyms.context import set_context
>>> set_context(device_target="GPU")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/dl/.conda/envs/tinyms/lib/python3.7/site-packages/mindspore/_checkparam.py", line 833, in wrapper
    return func(*args, **kwargs)
  File "/home/dl/.conda/envs/tinyms/lib/python3.7/site-packages/mindspore/context.py", line 662, in set_context
    raise ValueError(f"Error, package type {__package_name__} support device type {__device_target__}, "
ValueError: Error, package type mindspore support device type cpu, but got device target GPU

我已经查看了文档，发现根本就没说如何安装 GPU 版本的 TinyMS。
（如果是我漏看了，请指正）

[songquanpeng]

解决了，貌似是因为使用 Conda 安装的 CUDA 不行，手动安装后解决了该问题。

另外也可能和 mindspore-gpu 和 tinyms 的安装顺序有关。