Added the [and-gpu] flag to tensorflow to hopefully utilize the cluster gpu
This commit is contained in:
+1
-1
@@ -43,7 +43,7 @@ dependencies:
|
|||||||
- six==1.17.0
|
- six==1.17.0
|
||||||
- tensorboard==2.19.0
|
- tensorboard==2.19.0
|
||||||
- tensorboard-data-server==0.7.2
|
- tensorboard-data-server==0.7.2
|
||||||
- tensorflow==2.19.0
|
- tensorflow[and-cuda]==2.19.0
|
||||||
- tensorflow-io-gcs-filesystem==0.37.1
|
- tensorflow-io-gcs-filesystem==0.37.1
|
||||||
- termcolor==3.1.0
|
- termcolor==3.1.0
|
||||||
- threadpoolctl==3.6.0
|
- threadpoolctl==3.6.0
|
||||||
|
|||||||
Reference in New Issue
Block a user