search

BVLC / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
34.09k stars 18.7k forks source link

Possibility to support Win10+Python2.7+CUDA10 #6766

Open momo1986 opened 5 years ago

momo1986 commented 5 years ago

Hello,

My operation system: Win10 My python version: 2.7 My VS version: 2015 My CUDA version: CUDA10.0.

To be compatible with other deep learning framework such as Tensorflow, Pytorch and MXNET. I install with CUDA10.0.

But when I build, it reports the error: caffeproto.vcxproj -> D:\caffe\scripts\build\lib\Release\caffeproto.lib Building NVCC (Device) object src/caffe/CMakeFiles/cuda_compile_1.dir/layers/Release/cuda_compile_1_generatedabsval layer.cu.obj nvcc fatal : Unsupported gpu architecture 'compute_481' CMake Error at cuda_compile_1_generated_absval_layer.cu.obj.Release.cmake:219 (message): Error generating D:/caffe/scripts/build/src/caffe/CMakeFiles/cuda_compile_1.dir/layers/Release/cuda_compile_1_generated_absval_layer .cu.obj

Is there possibility to install with CUDA 10.0 or should I roll back the version to CUDA 8.0 to support Caffe?

How can I resolve this error?

Regards!

400lbhacker commented 5 years ago

hello, have a look on my page for precompiled (pre built) caffe.... supports 64x Windows, python 2.7