首页 > 编程语言 > 详细

ubuntu 自动切换cudnn python 版本

时间:2019-09-17 11:08:32      阅读:102      评论:0      收藏:0      [点我收藏+]

需要文件夹cuda8.0-cudnn5, cuda8.0-cudnn6
switch_cudnn.sh

#usr/bin/env sh
VERSION=$1
cd /media/data_1/Yang/software_install/cudnn/cuda8.0-cudnn${VERSION}
sudo cp ./include/cudnn.h /usr/local/cuda/include
sudo cp ./lib64/lib* /usr/local/cuda/lib64/  ##复制动态链接库
cd /usr/local/cuda/lib64/

sudo ln -snf libcudnn.so.${VERSION} libcudnn.so    ##强制建立软链接

echo 'cudnn.__version__='${VERSION}

运行: ./switch_cudnn.sh 5 ./switch_cudnn.sh 6

switch_python.sh

#usr/bin/env sh
VERSION=$1
sudo ln -snf /usr/bin/python${VERSION} /usr/bin/python
echo 'python.__version__='${VERSION}

ubuntu 自动切换cudnn python 版本

原文:https://www.cnblogs.com/yanghailin/p/11531551.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!