欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu下常用配置笔记(一)Cuda版本切换

程序员文章站 2022-07-04 21:09:48
...

Cuda版本切换

感谢这个大佬,本模块参考自
https://blog.csdn.net/yinxingtianxia/article/details/80462892
以 cuda_8.0.61_375.26_linux.run为例
cd进入 cuda_8.0.61_375.26_linux.run

sudo bash cuda_8.0.61_375.26_linux.run #开始安装

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
 y)es/(n)o/(q)uit: n  #是否显卡驱动包,由于已经安装显卡驱动,选择n 
 Install the CUDA 8.0 Toolkit?
 (y)es/(n)o/(q)uit: y #是否安装工具包,选择y 
 Enter Toolkit Location
 [ default is /usr/local/cuda-8.0 ]: #工具包安装地址,默认回车即可 
 Do you want to install a symbolic link at /usr/local/cuda?
 (y)es/(n)o/(q)uit: n #注意这个是否添加软链接的选项,如果你这个时第二个cuda版本,且不希望立即转换为这个版本,则就要选n
————————————————

注意bashrc文件的配置

vim ~/.bashrc #修改配置文件
# 配置处应该是
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
# 注意这里的cuda不要是cuda-8.0、cuda-9.0之类的,cuda一个软连接的中介,我们之后切换就是通过这个

查看/usr/local/路径,可以看到现在已经装好的版本,如cuda-10.0,cuda-9.0,cuda-8.0等
使用以下命令可以查看目前cuda软链接指向哪个版本,即现在使用的版本

cd /usr/local #进入路径
stat cuda

如图:
Ubuntu下常用配置笔记(一)Cuda版本切换
最后切换版本只需要先删除cuda软连接,然后再重新连接到想要的版本上就行

sudo rm -rf cuda
sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda

其他命令

nvcc -V #查看cuda版本信息

cuDNN安装

官网下载好对应的cudnn版本

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp cuda/lib64/* /usr/local/cuda-8.0/lib64/

tensorflow安装

anaconda下虚拟环境安装

建立conda虚拟环境

conda create -n tensorflow python=2.7

**虚拟环境

source activate teensorflow

安装tensorflow(在虚拟环境下)

pip install tensorflow==1.2.1