diff --git a/paddle/scripts/fast_install.sh b/paddle/scripts/fast_install.sh index 287534cd0c970fe4078fcb83ecab50c96270a7b6..32dccd258f804369b599962419c268935db4829d 100644 --- a/paddle/scripts/fast_install.sh +++ b/paddle/scripts/fast_install.sh @@ -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