bs
/
Step_Data_Project_India
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Added the [and-gpu] flag to tensorflow to hopefully utilize the cluster gpu

master
Robert Rabbe 6 days ago
parent
1e85d5f3e2
commit
4f4e9244ab
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      environment.yaml

2
environment.yaml
View File

@ -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

Write Preview
Loading…
Cancel
Save