fix linux bug

revert-15296-async_double_buffered_py_reader
tianshuo78520a 6 years ago
parent 3308e3c4cb
commit 25c032bb2c

@ -346,17 +346,17 @@ function PipLinuxInstall(){
if [[ "$paddle_version" == "2" ]];then
if [[ "$GPU" == "gpu" ]];then
if [[ ${AVX} == "avx" ]];then
rm -rf `echo $wheel_cpu_develop|awk -F '/' '{print $NF}'`
wget $wheel_cpu_develop
rm -rf `echo $wheel_gpu_release|awk -F '/' '{print $NF}'`
wget $wheel_gpu_release
$pip_path install --user -i https://mirrors.aliyun.com/pypi/simple --trusted-host=mirrors.aliyun.com $wheel_gpu_release
else
rm -rf `echo $wheel_cpu_release_nvax|awk -F '/' '{print $NF}'`
wget $wheel_cpu_release_nvax
rm -rf `echo $wheel_gpu_release_novax|awk -F '/' '{print $NF}'`
wget $wheel_gpu_release_novax
$pip_path install --user -i https://mirrors.aliyun.com/pypi/simple --trusted-host=mirrors.aliyun.com $wheel_gpu_release_noavx
fi
else
rm -rf `echo $wheel_cpu_develop|awk -F '/' '{print $NF}'`
wget $wheel_cpu_develop
rm -rf `echo $wheel_cpu_release|awk -F '/' '{print $NF}'`
wget $wheel_cpu_release
$pip_path install --user -i https://mirrors.aliyun.com/pypi/simple --trusted-host=mirrors.aliyun.com $wheel_cpu_release
fi
else
@ -375,8 +375,8 @@ function PipLinuxInstall(){
function checkLinuxGPU(){
AVX=`cat /proc/cpuinfo |grep avx|tail -1|grep avx`
which_gpu=`lspci |grep -i nvidia`
if [ "$which_gpu" == "" ];then
which nvidia-smi >/dev/null 2>&1
if [ "$?" != "0" ];then
GPU='cpu'
echo "您使用的是不包含支持的GPU的机器"
else

Loading…
Cancel
Save