This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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( |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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( |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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( |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
(ros_env) C:\Users\songyucheng>mamba install vs2022_win-64 | |
__ __ __ __ | |
/ \ / \ / \ / \ | |
/ \/ \/ \/ \ | |
███████████████/ /██/ /██/ /██/ /████████████████████████ | |
/ / \ / \ / \ / \ \____ | |
/ / \_/ \_/ \_/ \ o \__, | |
/ _/ \_____/ ` | |
|/ |