Skip to content

bonlime/pytorch-tools

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch-Tools

Tool box for PyTorch for fast prototyping.

Overview

Installation

Requires GPU drivers and CUDA already installed.

pip install git+https://github.com/bonlime/pytorch-tools.git@master

It is also recommended to install NVIDIA Apex to allow usage of additional optimizers

pip install ---upgrade -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git

Designed and maintained by @bonlime and @zakajd

Star the project if you like it!