Comments (5)
What GPU are you using?
So what happens is that PyTorch tries to map the tensors in the checkpoint directly on the GPU and docker cannot see any GPUs which is really strange. In the meantime try the fix in the last line of the error message, i.e. pass map_location='cpu'
in torch.load
from spin.
Ok, apparently the package future is not in the docker image but I had it installed locally so it was working. I updated the image. Can you try again now? (make sure to pull the image again)
from spin.
Hi, thank you for the quick reply, I pull the updated docker.
Now I think there is a issue with pyrender dependency freetype and pyglet, I installing those using pip.
run fetch_data.sh and using nvidia-docker run chaneyk/spin -it run the demo
I get the following error
Traceback (most recent call last):
File "demo.py", line 105, in
checkpoint = torch.load(args.checkpoint)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 387, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 574, in _load
result = unpickler.load()
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 537, in persistent_load
deserialized_objects[root_key] = restore_location(obj, location)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 119, in default_restore_location
result = fn(storage, location)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 95, in _cuda_deserialize
device = validate_cuda_device(location)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 79, in validate_cuda_device
raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location='cpu' to map your storages to the CPU.
Indeed while in the container created by chaneyk/spin - nvidia-smi is empty
While the docker works with gpu fine:
nvidia-docker run nvidia/cuda:9.0-base nvidia-smi shows all the gpu
from spin.
Update: You need to run nvidia-docker2
from spin.
When using map_location='cpu' and renaming the smpl neutral model:
mv ./data/smpl/basicModel_neutral_lbs_10_207_0_v1.0.0.pkl ./data/smpl/SMPL_NEUTRAL.pkl
It works
Thank you
from spin.
Related Issues (20)
- Question about camera parameters in order to render in Pytorch3d HOT 3
- about dataset preprocess for mpi-inf-3dhp
- A bug in 'mpi_inf_3dhp.py'
- 2d joints for original image
- final fits npz file link not working HOT 5
- Differnece between unaligned and aligned H3.6m moshed data?
- Project
- How can I generate openpose json for MPII?
- Questions about data augmentation
- RuntimeError: size of dimension does not match previous size, operand 1, dim 2
- Some error I got HOT 2
- pck
- AssertionError: Path ./body_models/smpl/SMPL_NEUTRAL.pkl does not exist! HOT 2
- Some problems about body shape
- The formula for camera translation
- About reimplement HOT 1
- About performance HOT 2
- About the parameters of SMPL poses and camera pose
- Error Omesa
- How to use demo.py to get more result? Like Mesh result ...
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 spin.