search

autonomousvision / occupancy_networks

This repository contains the code for the paper "Occupancy Networks - Learning 3D Reconstruction in Function Space"
https://avg.is.tuebingen.mpg.de/publications/occupancy-networks
MIT License
1.48k stars 291 forks source link

Please help me,i have a problem about cuda and pytorch version #100

Open kunzaiyy opened 3 years ago

kunzaiyy commented 3 years ago

After i installed cuda( conda install pytorch torchvision cudatoolkit=9.0 -c pytorch),when i ran python generate.py configs/demo.yaml,i got this problem:: RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1556653183467/work/aten/src/THC/THCBlas.cu:259

I inquired from Google and they told me that cuda9.0 is not supported by the graphics card, but when I applied cuda10.0, this would happen：： ImportError: /home/kun/anaconda3/envs/mesh_funcspace/lib/python3.6/site-packages/torch/lib/libmkldnn.so.0: undefined symbol: cblas_sgemm_alloc

So,how did you run this code successfully，please tell me，my system version is ubuntu18.04,thank you very much!!

BQT11 commented 2 years ago

I also have the same problem.

dnvtmf commented 2 years ago

solved by run command "conda install mkl -c anaconda"