当前位置:   article > 正文

cuda 环境变量设置 和 cudnn拷贝_如何复制一台机器的cuda环境

如何复制一台机器的cuda环境

多个cuda版本的切换参考如下文章
https://blog.csdn.net/yinxingtianxia/article/details/80462892

比较重要的两点是:

安装过程中在建立软链接时需要注意一下,如果你是第一次安装cuda,那么毫无疑问输入y(yes),但是如果你是安装额外版本的cuda,是否选择y(yes)就要看你的具体需求而定,简言之,就是如果你希望启用当前安装的cuda版本,就选y,如果你只是想安装这个版本,而暂时还不想启用该版本时,就选n

可以看到,文件类型是symbolic link,而指向的目录正是/usr/local/cuda-9.0,当我们想使用cuda-9.1版本时,只需要删除该软链接,然后重新建立指向cuda-9.1版本的软链接即可(注意名称还是cuda,因为要与bashrc文件里设置的保持一致)

切换版本方法


sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda
  • 1
  • 2
  • 3

下载完cudnn9.0之后进行解压,cd进入解压之后的include目录,在命令行进行如下操作

sudo cp cudnn.h /usr/local/cuda/include/    #复制头文件
sudo cp lib* /usr/local/cuda/lib64/    #复制动态链接库
  • 1
  • 2

(设置环境变量时,首先确定好cuda安装路径和位置,这一步非常重要,在安装时不需要对下面位置进行修改,系统会自动建立连接)

在/etc/profile 或者 ~/.bashrc文件 中添加CUDA环境变量(最好使用vim,不要用gedit,刚刚装完系统gedit会导致键盘错乱,增加不必要的麻烦)

sudo gedit /etc/profile

在最后行添加内容:

export PATH=/usr/local/cuda/bin:$PATH  
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
  • 1
  • 2

保存后, 执行下列命令, 使环境变量立即生效

source /etc/profile

同时需要添加lib库路径: 在 /etc/ld.so.conf.d/加入文件 cuda.conf,

sudo vim /etc/ld.so.conf.d/cuda.conf

内容如下

/usr/local/cuda/lib64
/lib
/usr/lib
/usr/lib32

这里把lib库都加全了

保存后,执行下列命令使之立刻生效

sudo ldconfig -v

2.2 安装CUDA SAMPLE

进入/usr/local/cuda/samples, 执行下列命令来build samples

sudo make all -j8

全部编译完成后, 进入/usr/local/cuda/samples/1_Utilities/deviceQuery$ ./deviceQuery , 运行deviceQuery

/usr/local/cuda/samples/1_Utilities/deviceQuery$ ./deviceQuery

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/151246
推荐阅读
相关标签
  

闽ICP备14008679号