Docker: paddlepaddle/paddle:latest-gpu-cuda9.0-cudnn7, it works in my computer, but when I install "pip install paddlepaddle-gpu", it does not work in my computer.
Created by: hdjsjyl
Docker: paddlepaddle/paddle:latest-gpu-cuda9.0-cudnn7, it works in my computer, but when I install "pip install paddlepaddle-gpu", it does not work in my computer. It shows the following error:
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)