songyuc

(ros_env) C:\Users\songyucheng>mamba install vs2022_win-64

                  __    __    __    __
                 /  \  /  \  /  \  /  \
                /    \/    \/    \/    \
███████████████/  /██/  /██/  /██/  /████████████████████████
              /  / \   / \   / \   / \  \____
             /  /   \_/   \_/   \_/   \    o \__,
            / _/                       \_____/  `
            |/

songyuc

cmake_minimum_required(VERSION 3.23)

set(CMAKE_CUDA_COMPILER "/usr/local/cuda/bin/nvcc")
# NECESSARY: because we need to specify the CUDA_COMPILER explicitly.
if (NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
    set(CMAKE_CUDA_ARCHITECTURES "75;80;86") # native is supported in version 3.24
endif ()
# Note: This variable is used to initialize the CUDA_ARCHITECTURES property on all targets.

project(Cuda_notes LANGUAGES CUDA CXX C)

songyuc

import nvidia.dali.fn as fn
from nvidia.dali import pipeline_def
from nvidia.dali.plugin.pytorch import DALIGenericIterator
from tqdm import tqdm


@pipeline_def
def coco_pipeline():
    coco_root = "/local/COCO"
    images, bounding_boxes, labels, image_ids = fn.readers.coco(

songyuc

import nvidia.dali.fn as fn
from nvidia.dali import pipeline_def
from nvidia.dali.plugin.pytorch import DALIGenericIterator
from tqdm import tqdm


@pipeline_def
def coco_pipeline():
    coco_root = "/local/COCO"
    images, bounding_boxes, labels, image_ids = fn.readers.coco(

songyuc

import nvidia.dali.fn as fn
from nvidia.dali import pipeline_def
from nvidia.dali.plugin.pytorch import DALIGenericIterator
from tqdm import tqdm


@pipeline_def
def coco_pipeline():
    coco_root = "/***/COCO"
    images, bounding_boxes, labels, image_ids = fn.readers.coco(