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