Error on Apple M1 pro
Open
bug:confirmed
topic: TensorFlow
- Dominant language
- Python
- Stars
- 14.3k
- Forks
- 3.1k
- PR merge metrics
- No merged PRs in 30d
Description
# KeyError: 'mps'
When I tried to use DGLGraph.to('mps'), a "keyerror" was issued.
# Data error when use tensorflow
When I tried to load the Cora dataset, the following error occurred:
“_dgl._ffi.base.DGLError: Cannot assign node feature "train_mask" on device /gpu:0 to a graph on device /cpu:0. Call DGLGraph.to() to copy the graph to the same device._”
Note: I install the latest version of dgl (0.9.0), pytorch (1.12.0) and tensorflow-metal(2.9).
How can I solve this problem.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.