# A implementation of ResNet-18 uses lot of GPU memory

**URL:** https://discourse.julialang.org/t/a-implementation-of-resnet-18-uses-lot-of-gpu-memory/36389
**Category:** Machine Learning
**Tags:** question, flux
**Created:** [March 23, 2020, 12:08pm UTC](https://discourse.julialang.org/t/a-implementation-of-resnet-18-uses-lot-of-gpu-memory/36389 "2020-03-23T12:08:50Z")
**Posts on this page:** 1
**Showing post:** 6

<div class="post-metadata">

### Author: ![OTapio](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/otapio/32/13656_2.png) [@OTapio](https://discourse.julialang.org/u/OTapio)
#### Post date: [March 26, 2020, 2:32pm UTC](https://discourse.julialang.org/t/a-implementation-of-resnet-18-uses-lot-of-gpu-memory/36389/6 "2020-03-26T14:32:28Z")

</div>

I’m using RTX 2070. It took me 6min in total. My Tensorflow output in full can be seen here [https://pastebin.com/qa1Zgft3](https://pastebin.com/qa1Zgft3)

If I use same GPU metrics as you have above I get following results

 ![gpu_usage_julia_flux2](https://global.discourse-cdn.com/julialang/original/3X/9/d/9d79d5b82edf737cabd5c57c7405d3b3ac494bc5.png)

When I use Tensorflow my GPU metrics are as follows

 ![gpu_usage_tensorflow](https://global.discourse-cdn.com/julialang/original/3X/6/3/63302533c279872ca64352fbf3f1a30f6df08f84.png)

I was given a hint that this might help me [https://juliagpu.gitlab.io/CUDA.jl/development/profiling/#Application-profiling-1](https://juliagpu.gitlab.io/CUDA.jl/development/profiling/#Application-profiling-1) . I haven’t had time to try it properly.

---

_[View the full topic](https://discourse.julialang.org/t/a-implementation-of-resnet-18-uses-lot-of-gpu-memory/36389)._