rapidsai / cuxfilter

GPU accelerated cross filtering with cuDF.
https://docs.rapids.ai/api/cuxfilter/stable/
Apache License 2.0
272 stars 67 forks source link

[BUG] Environment compatability issue with Tensorflow 2.12 #511

Closed exactlyallan closed 1 year ago

exactlyallan commented 1 year ago

Per @beckernick

mamba create -n cuxfilter-23.08-tf-py310 -c rapidsai-nightly -c nvidia -c conda-forge cuxfilter=23.08 python=3.10 cudatoolkit=11.8 tensorflow=2.12

Looking for: ['cuxfilter=23.08', 'python=3.10', 'cudatoolkit=11.8', 'tensorflow=2.12']

nvidia/linux-64 Using cache nvidia/noarch Using cache conda-forge/linux-64 Using cache conda-forge/noarch Using cache rapidsai-nightly/noarch [====================] (00m:00s) No change rapidsai-nightly/linux-6 [====================] (00m:00s) No change pkgs/main/linux-64 [====================] (00m:00s) No change pkgs/r/linux-64 [====================] (00m:00s) No change pkgs/main/noarch [====================] (00m:00s) No change pkgs/r/noarch [====================] (00m:00s) No change Encountered problems while solving:

See below for system, environment creation, and smoke test information:

Smoke Test Environments

Tensorflow mamba create -n rapids-23.08-tf-py39 -c rapidsai-nightly -c nvidia -c conda-forge rapids=23.08 python=3.9 cudatoolkit=11.8 tensorflow mamba create -n rapids-23.08-tf-py310 -c rapidsai-nightly -c nvidia -c conda-forge rapids=23.08 python=3.10 cudatoolkit=11.8 tensorflow

PyTorch mamba create -n rapids-23.08-pytorch-py39 -c rapidsai-nightly -c pytorch -c nvidia -c conda-forge rapids=23.08 python=3.9 cudatoolkit=11.8 pytorch torchvision torchaudio pytorch-cuda=11.8 mamba create -n rapids-23.08-pytorch-py310 -c rapidsai-nightly -c pytorch -c nvidia -c conda-forge rapids=23.08 python=3.10 cudatoolkit=11.8 pytorch torchvision torchaudio pytorch-cuda=11.8

Smoke Tests

PyTorch import cudf; import cuml; import cugraph; import cuspatial; import torch; torch.cuda.is_available()

Tensorflow import cudf; import cuml; import cugraph; import cuspatial; import tensorflow as tf; tf.config.list_physical_devices("GPU")

System Ubuntu 20.04 CUDA Driver 520.61.05 CUDA Version 11.8 RTX 8000 GPU

AjayThorve commented 1 year ago

This is now fixed, the above command

mamba create -n cuxfilter-23.08-tf-py310 -c rapidsai-nightly -c nvidia -c conda-forge cuxfilter=23.08 python=3.10 cudatoolkit=11.8 tensorflow=2.12

now works. @exactlyallan