wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 20:53:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Docker] Use Cuda 11 (#13691)

Browse Source
This commit is contained in:
Ian Rodney
2021-01-27 13:45:30 -08:00
committed by GitHub
parent eba698d48e
commit b4bcb9b60a
6 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/requirements_ml_docker.txt
+5 -2
View File
@@ -1,3 +1,6 @@
ipython
tensorflow-gpu
torch
tensorflow-gpu>=2.4.0
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.7.1+cu110
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.8.2+cu110