常见问答
发布于2021-12 浏览:2051 回复:0
0
收藏
快速回复

1 在ai studio或者notebook下载cuda
!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run

2 新建一个目录~/cuda-9.0/

3 将下载的cuda安装到上述新建的目录中
!sh cuda_9.0.176_linux-run --silent --toolkit --toolkitpath=$HOME/cuda-9.0

4 下载cudnn,注意这个需要去官网注册账号,事先下载到自己的电脑然后,
注意版本要对,然后通过新建数据集上传到ai studio的data/目录下,名字尽量短,下载之后是一个tgz格式 的文件,我把名字改为cudnn-9.0.tgz

5 解压4步下载的文件到根目录,解压之后的cudnn文件名默认为cuda,
!tar -zxvf /home/aistudio/data/data25688/cudnn-9.0.tgz

6 解压把cudnn的指定文件copy到cuda安装文件对应的目录中注意目录要对,这一步只需要做一次就可以
!cp cuda/include/cudnn.h cuda-9.0/include/
!cp cuda/lib64/libcudnn* cuda-9.0/lib64/

至此准备工作完成,这些工作只需要一次就可以,接下来进行环境配置阶段
————————————————
版权声明:本文为CSDN博主「月色很柔」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_41971768/article/details/105645906

收藏
点赞
0
个赞
TOP
切换版块