Installing paddle for CUDA9 and face detection errors
Created by: hdjsjyl
I use pip install paddlepaddle-gpu for CUDA9 to install paddlepaddle. But it shows I have installed paddlepaddle-gpu 1.14.0post87, which is for CUDA8, why it happens? When I run train.py
Traceback (most recent call last): File "train.py", line 178, in train(args, config, train_file_list, optimizer_method="momentum") File "train.py", line 52, in train sub_network=use_pyramidbox) File "/home/daguo/code/PaddlePaddle_models/fluid/face_detection/pyramidbox.py", line 76, in init self._pyramidbox() File "/home/daguo/code/PaddlePaddle_models/fluid/face_detection/pyramidbox.py", line 243, in _pyramidbox face_loc = permute_and_reshape(face_loc, 4) File "/home/daguo/code/PaddlePaddle_models/fluid/face_detection/pyramidbox.py", line 227, in permute_and_reshape np.array([0, -1, last_dim]).astype("int32")) TypeError: assign() takes exactly 2 arguments (1 given)