Comments (5)
FYI.
Google Cloud ML Engine already has an optimized and GPU-enabled TensorFlow version installed. you could remove the tensorflow line from setup.py to make it work,or the GPU not being used if you download the CPU-only tensorflow package from PyPI and shadowing the existing TensorFlow installation.
from sonnet.
@Shuolongbj Hi, i make Sonnet work on google Cloud after a while, i did it by compiling the library on my machine, upload the wheel file to Google Cloud Storage, and specifying the --package option when uploading the job.
After a while I came to realize that the trainign was going slow, so I started to think that the gpu was not being used. what i think, needs to be done is recompiling the sonnet library against a GPU supported Tensorflow, and it maybe work fine in that case.
Thanks for your answer. :)
from sonnet.
Hi @afhuertass,
Sonnet should be using the version of tensorflow you installed with pip.
Do you get only CPU devices when you run a session with device placement being logged?
(see here)
And do you get GPU devices when you run tensorflow ops without Sonnet?
from sonnet.
@afhuertass Can you try installing the pip package (either 'dm-sonnet' or 'dm-sonnet-gpu') ?
from sonnet.
Closed due to inactivity. Feel free to reopen the bug if you have any problem installing from pip.
from sonnet.
Related Issues (20)
- Understanding how to use mixed precision training HOT 2
- Conv2D initialize error in tf function
- Typo HOT 1
- this portion of attention code looks incorrect
- Cannot install sonnet on mac m1 HOT 5
- Decaying learning rate HOT 1
- Can you explain the logic of updated_ema_cluster_size in VectorQuantizerEMA?
- Convert a Sonnet model to a Keras model HOT 1
- (Maybe) in-consistency between VQ-VAE paper and its implementation. HOT 2
- Regarding LearningRateSchedule for ADAM optimizer HOT 2
- Learning rate and optimizer used for the VQ-VAE-2 paper HOT 2
- I have a question about the perplexity term (in the VQ-VAE). HOT 2
- Issues using Sonnet HOT 3
- How to save the checkpoint for self-defined model?
- TensorArray Not Used on line 71 of sgd_test.py HOT 2
- Is there a stereotypical way to visualize the network strucuture when using Sonnet? HOT 1
- sonnet abstract module in tensorflow2 HOT 10
- Laplace smoothing for EMA codebook update
- Basic import fails on Colab: cannot import name 'context' from 'tensorflow.python' HOT 11
- AttributeError: module 'nest' has no attribute 'Create' HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from sonnet.