Still having the same issue.
```
File "/home/carlo/.conda/envs/d2l/lib/python3.8/site-packages/mxnet/base.py",
line 255, in check_call
raise MXNetError(py_str(_LIB.MXGetLastError()))
mxnet.base.MXNetError: [14:20:17] src/operator/contrib/./../linalg_impl.h:213:
Check failed: e == CUBLAS_STATUS_SUCCESS (13 vs. 0) : cuBLAS:
CUBLAS_STATUS_EXECUTION_FAILED
```
The problem seems to be whenever I try to access the values of the loss
function. Is it necessary to do anything about that specifically?
Also, is it necessary to use:
```
X_batch = X_batch.as_in_context(ctx)
Y_batch = Y_batch.as_in_context(ctx)
```
considering that, at the beginning, I'm specifying the context for X_train,
y_train...?
```
y_train = mx.nd.array(y_train.to_numpy().reshape(-1,1), dtype=np.float32,
ctx=ctx)
y_test = mx.nd.array(y_test.to_numpy().reshape(-1,1), dtype=np.float32,
ctx=ctx)
X_train = mx.nd.array(X_train.to_numpy(), dtype=np.float32, ctx=ctx)
X_test = mx.nd.array(X_test.to_numpy(), dtype=np.float32, ctx=ctx)
```
---
[Visit
Topic](https://discuss.mxnet.apache.org/t/correct-way-to-train-sequential-model-on-gpu/6848/3)
or reply to this email to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.mxnet.apache.org/email/unsubscribe/9492b354b85cc22c8d63595a5b95d8d74b508eea474920f725c58f12725b01e6).
