提交 25c032bb 编写于 作者: T tianshuo78520a

fix linux bug

上级 3308e3c4
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册