https://github.com/speedcell4/torchdevice

Setup CUDA_VISIBLE_DEVICES
https://github.com/speedcell4/torchdevice

cuda deep-learning gpu machine-learning pytorch

Last synced: 4 days ago
JSON representation

Setup CUDA_VISIBLE_DEVICES

• Host: GitHub
• URL: https://github.com/speedcell4/torchdevice
• Owner: speedcell4
• Created: 2022-02-09T05:33:37.000Z (about 3 years ago)
• Default Branch: master
• Last Pushed: 2023-08-24T10:32:39.000Z (over 1 year ago)
• Last Synced: 2025-01-06T02:54:37.639Z (about 1 month ago)
• Topics: cuda, deep-learning, gpu, machine-learning, pytorch
• Language: Python
• Homepage:
• Size: 13.7 KB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        


# TorchDevice

![GitHub Workflow Status (with event)](https://img.shields.io/github/actions/workflow/status/speedcell4/torchdevice/unit-tests.yml?cacheSeconds=0)

![PyPI - Version](https://img.shields.io/pypi/v/torchdevice?label=pypi%20version&cacheSeconds=0)

![PyPI - Downloads](https://img.shields.io/pypi/dm/torchdevice?cacheSeconds=0)



Setup `CUDA_VISIBLE_DEVICES`.

## Installation

`python -m pip torchdevice`