Comments (14)
I deleted and reinstalled the entire project and finally it worked. Maybe I got problem when building.
I confirm that snake can be run on CUDA 0.1 + Pytorch 1.4.
Thanks for your support.
from snake.
What are the versions of your system cuda and pytorch cuda?
from snake.
The version of my system cuda is 10.2; my pytorch cuda is 10.1.
from snake.
Make sure two points:
- pytorch version that you used to compile the code should be the same as the one you use to run the code.
- The version of pytorch cuda should be the same as that of the system cuda.
from snake.
But I'm running your project on a server GPU which installed CUDA 10.2 already while the Pytorch official website only provide the introduction for installing CUDA upto 10.1.
Does anyone know how to install Pytorch with CUDA 10.2 from source? Thanks.
from snake.
Are you sure that the pytorch version that you used to compile the code is the same as the one you use to run the code?
from snake.
@pengsida how can I check these versions? Sorry, I'm just a newbie.
Maybe the pytorch version which I used to compile the code is 1.4.0 and cuda 10.1
When I run setup.py in apex, the terminal display something with these lines at the beginning:
torch.__version__ = 1.4.0
setup.py:46: UserWarning: Option --pyprof not specified. Not installing PyProf dependencies!
warnings.warn("Option --pyprof not specified. Not installing PyProf dependencies!")
Compiling cuda extensions with
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243
from /usr/local/cuda/bin
from snake.
python -c "import torch; print(torch.version.cuda)"
nvcc --version
from snake.
python -c "import torch; print(torch.version.cuda)"
nvcc --version
I ran these two command and recieved 2 responses, respectively:
10.1
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243
But when I ran the testing command, the error still happened. Any idea?
from snake.
How about re-compling this module: python setup.py build_ext --inplace
.
from snake.
How about re-compling this module:
python setup.py build_ext --inplace
.
I re-compiled dcn_v2, extreme_utils, roi_align_layer followed the INSTALL.md but the error still appeared.
from snake.
Did you set export CUDA_HOME="/usr/local/cuda-9.0
?
If your cuda is 10.0, it might cause errors.
from snake.
No. Depend on my GPU, I set CUDA_HOME="/usr/local/cuda".
from snake.
You could test pytorch 1.1 and cuda 9.0, as described in INSTALL.md, to see if this problem is caused by the pytorch version and cuda version.
from snake.
Related Issues (20)
- Code errors and broken links HOT 1
- Mask map HOT 3
- 更换backbone HOT 4
- 修改loss函数 HOT 1
- 对比实验评估问题 HOT 1
- Cityscapes数据集缺失问题
- Cityscapes数据集缺失问题 HOT 2
- “THC/THC.h”: No such file or directory HOT 5
- dla-34改dla60 HOT 1
- RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED HOT 2
- SBD数据集貌似失效了,请问可以更新一下链接嘛! HOT 1
- ex_loss 和py_loss的含义 HOT 1
- 使用自己制作COCO数据集报错的问题 HOT 1
- 如何能够在目标检测的框上面显示类别 HOT 2
- dance这篇论文在snake的基础上改进,修改了损失函数 HOT 1
- 作者您好,关于如何显示测试结果的分类别mAP? HOT 1
- 请问作者每一次迭代的效果图在代码中是怎么实现的 HOT 1
- 损失曲线问题
- Costum dataset
- windows10,python3.7,vs2022构建失败:dcn_v2,extreme_utils,roi_align_layer
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 snake.