Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Upstream anchor labels

CUDAGraph CUDAPluggableAllocator device Device device_of Event ExternalStream GdsFile GreenContext MemPool Stream StreamContext tensorplay.cuda.AcceleratorError tensorplay.cuda.caching_allocator_alloc tensorplay.cuda.caching_allocator_delete tensorplay.cuda.can_device_access_peer tensorplay.cuda.change_current_allocator tensorplay.cuda.check_error tensorplay.cuda.classproperty tensorplay.cuda.clock_rate tensorplay.cuda.CudaError tensorplay.cuda.cudart tensorplay.cuda.current_blas_handle tensorplay.cuda.current_device tensorplay.cuda.current_solver_handle tensorplay.cuda.current_stream tensorplay.cuda.default_stream tensorplay.cuda.DeferredCudaCallError tensorplay.cuda.device_count tensorplay.cuda.device_memory_used tensorplay.cuda.empty_cache tensorplay.cuda.get_allocator_backend tensorplay.cuda.get_arch_list tensorplay.cuda.get_device_capability tensorplay.cuda.get_device_name tensorplay.cuda.get_device_properties tensorplay.cuda.get_gencode_flags tensorplay.cuda.get_per_process_memory_fraction tensorplay.cuda.get_stream_from_external tensorplay.cuda.get_sync_debug_mode tensorplay.cuda.graph_annotations.clear_kernel_annotations tensorplay.cuda.graph_annotations.get_kernel_annotations tensorplay.cuda.graph_annotations.is_available tensorplay.cuda.graph_annotations.mark_kernels tensorplay.cuda.graphs.export_dot tensorplay.cuda.graphs.export_graph_data tensorplay.cuda.graphs.graph tensorplay.cuda.graphs.graph_pool_handle tensorplay.cuda.graphs.is_current_stream_capturing tensorplay.cuda.graphs.make_graphed_callables tensorplay.cuda.host_memory_stats tensorplay.cuda.host_memory_stats_as_nested_dict tensorplay.cuda.init tensorplay.cuda.ipc_collect tensorplay.cuda.is_available tensorplay.cuda.is_bf16_supported tensorplay.cuda.is_gds_available tensorplay.cuda.is_initialized tensorplay.cuda.is_tf32_supported tensorplay.cuda.list_gpu_processes tensorplay.cuda.max_memory_allocated tensorplay.cuda.max_memory_cached tensorplay.cuda.max_memory_reserved tensorplay.cuda.mem_get_info tensorplay.cuda.memory_allocated tensorplay.cuda.memory_cached tensorplay.cuda.memory_reserved tensorplay.cuda.memory_snapshot tensorplay.cuda.memory_stats tensorplay.cuda.memory_stats_as_nested_dict tensorplay.cuda.memory_summary tensorplay.cuda.memory_usage tensorplay.cuda.memory.caching_allocator_disabled tensorplay.cuda.memory.caching_allocator_enable tensorplay.cuda.memory.use_mem_pool tensorplay.cuda.nccl.is_available tensorplay.cuda.nccl.version tensorplay.cuda.nvtx.mark tensorplay.cuda.nvtx.range tensorplay.cuda.nvtx.range_end tensorplay.cuda.nvtx.range_pop tensorplay.cuda.nvtx.range_push tensorplay.cuda.nvtx.range_start tensorplay.cuda.OutOfMemoryError tensorplay.cuda.power_draw tensorplay.cuda.profiler.profile tensorplay.cuda.profiler.start tensorplay.cuda.profiler.stop tensorplay.cuda.random.get_rng_state tensorplay.cuda.random.get_rng_state_all tensorplay.cuda.random.initial_seed tensorplay.cuda.random.manual_seed tensorplay.cuda.random.manual_seed_all tensorplay.cuda.random.seed tensorplay.cuda.random.seed_all tensorplay.cuda.random.set_rng_state tensorplay.cuda.random.set_rng_state_all tensorplay.cuda.reset_accumulated_host_memory_stats tensorplay.cuda.reset_accumulated_memory_stats tensorplay.cuda.reset_max_memory_allocated tensorplay.cuda.reset_max_memory_cached tensorplay.cuda.reset_peak_host_memory_stats tensorplay.cuda.reset_peak_memory_stats tensorplay.cuda.set_device tensorplay.cuda.set_per_process_memory_fraction tensorplay.cuda.set_stream tensorplay.cuda.set_sync_debug_mode tensorplay.cuda.stream tensorplay.cuda.synchronize tensorplay.cuda.temperature tensorplay.cuda.utilization

API symbolscuda

Copy

View Markdown Download .md

tensorplay.cuda.nccl.is_available

tensorplay.cuda.nccl.is_available(tensors) → bool[source]

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.cuda.memory.use_mem_pool

Route allocations to a MemPool (not supported by this build).

tensorplay.cuda.nccl.version

Returns the version of the NCCL.