fix(env): fix some potiential env issue later

* add some notes

Author: Kin-Zhang

assets/README.md

@@ -80,6 +80,7 @@ python -c "import torch; print(torch.__version__); print(torch.cuda.is_available
 python -c "import lightning.pytorch as pl; print(pl.__version__)"
 python -c "from assets.cuda.mmcv import Voxelization, DynamicScatter;print('successfully import on our lite mmcv package')"
 python -c "from assets.cuda.chamfer3D import nnChamferDis;print('successfully import on our chamfer3D package')"
+python -c "from av2.utils.io import read_feather; print('av2 package ok')"
 ```
 
 

assets/cuda/chamfer3D/setup.py

@@ -1,15 +1,25 @@
 from setuptools import setup
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
+extra_compile_args = {
+    'cxx': ['-DCCCL_IGNORE_DEPRECATED_CUDA_BELOW_12'],
+    'nvcc': ['-DCCCL_IGNORE_DEPRECATED_CUDA_BELOW_12'],
+}
+
 setup(
     name='chamfer3D',
     ext_modules=[
-        CUDAExtension('chamfer3D', [
-            "/".join(__file__.split('/')[:-1] + ['chamfer3D_cuda.cpp']), # must named as xxx_cuda.cpp
-            "/".join(__file__.split('/')[:-1] + ['chamfer3D.cu']),
-        ]),
+        CUDAExtension(
+            name='chamfer3D',
+            sources=[
+                "/".join(__file__.split('/')[:-1] + ['chamfer3D_cuda.cpp']), # must named as xxx_cuda.cpp
+                "/".join(__file__.split('/')[:-1] + ['chamfer3D.cu']),
+            ],
+            extra_compile_args=extra_compile_args
+        ),
     ],
     cmdclass={
         'build_ext': BuildExtension
     },
-    version='1.0.1')
+    version='1.0.2'
+)

assets/cuda/mmcv/setup.py

@@ -3,9 +3,15 @@
 from setuptools import find_packages, setup
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
+extra_compile_args = {
+    'cxx': ['-DCCCL_IGNORE_DEPRECATED_CUDA_BELOW_12'],
+    'nvcc': ['-DCCCL_IGNORE_DEPRECATED_CUDA_BELOW_12'],
+}
+
+
 setup(
     name='mmcv',
-    version='1.0.1',
+    version='1.0.2',
     ext_modules=[
         CUDAExtension(
             name='mmcv',
@@ -18,14 +24,8 @@
                 "/".join(__file__.split("/")[:-1] + ["pybind.cpp"]),
 
             ],
-            # extra_compile_args={
-            #     'cxx': ['-std=c++17'], 
-            #     'nvcc': ['-std=c++17',
-            #     '-D__CUDA_NO_HALF_OPERATORS__',
-            #     '-D__CUDA_NO_HALF_CONVERSIONS__',
-            #     '-D__CUDA_NO_HALF2_OPERATORS__',
-            #              ],}
-            ),
+            extra_compile_args=extra_compile_args
+        ),
     ],
     cmdclass={'build_ext': BuildExtension},
 

environment.yaml

@@ -34,6 +34,8 @@ dependencies:
     - assets/cuda/chamfer3D
     - assets/cuda/mmcv
     - open3d==0.18.0
+    # - opencv-python-headless==4.9.0.80 # for av2 if there is any on cv2....
+    - av==14.0.1 # need to specific this one otherwise av2 will have error
     - av2==0.3.1
     - spconv-cu118==2.3.6
     - numpy==1.26.4