https://github.com/ashwinpn/imgnet

Running NVIDIA profiler on the cloud
https://github.com/ashwinpn/imgnet

Last synced: about 6 hours ago
JSON representation

Running NVIDIA profiler on the cloud

• Host: GitHub
• URL: https://github.com/ashwinpn/imgnet
• Owner: ashwinpn
• Created: 2020-10-23T23:37:57.000Z (about 4 years ago)
• Default Branch: main
• Last Pushed: 2020-10-23T23:55:57.000Z (about 4 years ago)
• Last Synced: 2023-10-19T22:45:53.042Z (about 1 year ago)
• Language: Python
• Size: 7.81 KB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# imgnet

Running NVIDIA profiler on the cloud 

## Usage

1. Change runtime type selecting GPU as hardware accelerator

2. Git clone this repository:

```

!git clone https://github.com/ashwinpn/imgnet.git

```

3. Change permissions:

```

!chmod 755 imgnet/INSTALL.sh

``` 

4. Install cuda, nvcc, gcc, and g++:

```

!./imgnet/INSTALL.sh

```

5. Add `/usr/local/cuda/bin` to `PATH`:

```python

import os

os.environ['PATH'] += ':/usr/local/cuda/bin'

```

6. Run 

```

!nvprof python main.py -a alexnet -b 8 --epochs 1 --lr 0.01 

```