mirror of
https://github.com/vale981/ray
synced 2025-03-05 18:11:42 -05:00
84e958f330
* wip ' * upgrade requirements * add file * fix * fixes * Apply suggestions from code review Try mlagents==0.21.0 for now (works with torch 1.9). * Apply suggestions from code review * wip * wip * fix * fix * upgrade lightning bolts * address comment Co-authored-by: Sven Mika <sven@anyscale.io>
8 lines
268 B
Text
8 lines
268 B
Text
ipython
|
|
# In TF >v2, GPU support is included in the base package.
|
|
tensorflow==2.5.0
|
|
tensorflow-probability==0.13.0
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torch==1.9.0+cu111
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torchvision==0.10.0+cu111
|