install pyopencl

2020-11-04 11:22:11 -08:00
parent 3efb4f4df4
commit 0be3d2cdba
1 changed files with 1 additions and 1 deletions
@@ -112,7 +112,7 @@ class Tensor:

  def cuda(self):
    if not GPU:
-      raise Exception("No GPU Support")
+      raise Exception("No GPU Support, install pyopencl")
    if not self.gpu:
      require_init_gpu()
      assert self.data.dtype == np.float32   # only float32 on GPU