fix cu

Tokazama · Feb 12, 2019 · 3bbcab5 · 3bbcab5
1 parent 753767a
commit 3bbcab5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/games/pendulum/cuda/Manifest.toml b/games/pendulum/cuda/Manifest.toml
@@ -173,10 +173,10 @@ uuid = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
 version = "0.6.1"
 
 [[Gym]]
-deps = ["Flux", "JSExpr", "WebIO"]
-git-tree-sha1 = "3a5f79660ecbdaa200c94faf9f71639a3aa78932"
-repo-rev = "master"
-repo-url = "https://github.com/tejank10/Gym.jl.git"
+deps = ["CUDAnative", "CuArrays", "Flux", "JSExpr", "WebIO"]
+git-tree-sha1 = "00a013a3c8cb7f66e730f1b1d3df30b9a250eb4d"
+repo-rev = "GPU"
+repo-url = "https://github.com/tejank10/Gym.jl"
 uuid = "56b9baea-2481-11e9-37ae-75904354ad8c"
 version = "0.1.0"
 

diff --git a/games/pendulum/pendulum-diff_rl.jl b/games/pendulum/pendulum-diff_rl.jl
@@ -2,7 +2,7 @@ using Flux, Gym
 using Flux.Optimise: _update_params!
 using Statistics: mean
 using DataStructures: CircularBuffer
-using CuArrays
+#using CuArrays
 
 #Load game environment