NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.32k stars 1.38k forks source link

apex not support CUDA 10.1? #1010

Open PPPPWU opened 3 years ago

PPPPWU commented 3 years ago

Pytorch binaries were compiled with Cuda 10.1.243

running install running bdist_egg running egg_info writing apex.egg-info/PKG-INFO writing dependency_links to apex.egg-info/dependency_links.txt writing top-level names to apex.egg-info/top_level.txt reading manifest file 'apex.egg-info/SOURCES.txt' writing manifest file 'apex.egg-info/SOURCES.txt' installing library code to build/bdist.linux-x86_64/egg running install_lib running build_py running build_ext building 'fused_adam_cuda' extension gcc -pthread -B /root/miniconda3/envs/myconda/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/root/miniconda3/envs/myconda/lib/python3.7/site-packages/torch/include -I/root/miniconda3/envs/myconda/lib/python3.7/site-packages/torch/include/torch/csrc/api/include -I/root/miniconda3/envs/myconda/lib/python3.7/site-packages/torch/include/TH -I/root/miniconda3/envs/myconda/lib/python3.7/site-packages/torch/include/THC -I/usr/local/cuda/include -I/root/miniconda3/envs/myconda/include/python3.7m -c csrc/fused_adam_cuda.cpp -o build/temp.linux-x86_64-3.7/csrc/fused_adam_cuda.o -O3 -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=fused_adam_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11 cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++ csrc/fused_adam_cuda.cpp: In function ‘void adam(at::Tensor&, at::Tensor&, at::Tensor&, at::Tensor&, at::Tensor&, float, float, float, float, float, int, int, int, float)’: csrc/fused_adam_cuda.cpp:13:9: error: expected ‘;’ before ‘if’ if (p_copy.numel() > 0) CHECK_INPUT(p_copy); ^~ error: command 'gcc' failed with exit status 1

mingji123 commented 3 years ago

i have occured this error,the way i fixed it

find the file , csrc/fused_adam_cuda.cpp and add a ; and it just ok......

minwang-ai commented 3 years ago

i have occured this error,the way i fixed it

find the file , csrc/fused_adam_cuda.cpp and add a ; and it just ok......

Hi mingji, what did you mean of ''add a''?