Skip to content

Commit 9de1142

Browse files
authored
Install cuml-cu12 (#1459)
http://b/381287748
1 parent 5452732 commit 9de1142

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

kaggle_requirements.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# Please keep this in alphabetical order
2+
--extra-index-url https://pypi.nvidia.com
23
Altair>=5.4.0
34
Babel
45
Boruta
@@ -23,6 +24,7 @@ catboost
2324
category-encoders
2425
cesium
2526
comm
27+
cuml-cu12
2628
cytoolz
2729
dask-expr
2830
# Older versions of datasets fail with "Loading a dataset cached in a LocalFileSystem is not supported"

tests/test_cuml.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
class TestCuml(unittest.TestCase):
77
@gpu_test
88
@p100_exempt # b/342143152: cuML(>=24.4v) is inompatible with p100 GPUs.
9-
@unittest.skip("b/381287748 cuML is not installed in Colab.")
109
def test_pca_fit_transform(self):
1110
import unittest
1211
import numpy as np

0 commit comments

Comments
 (0)