我要装下mask rcnn(https://github.com/facebookresearch/maskrcnn-benchmark)
因为我现在服务器装了一个cuda10.1,然后我装的Pytorch又是基于cuda10.0版本的,所以我就编译不成功apex。只能在自己的目录下再装一个cuda10.0,因为不想用管理员账号装,怕破坏其他人的环境。反正经常要重复这些操作,而且每次都要百度,还不如自己写下这个过程,下次直接看~~~
事先准备工作:
去官网下载cuda安装包。
网址:下载地址
Step1:
然后,在安装包的目录下:
sh cuda_10.0.130_410.48_linux
出现的问题:我这文件明明在目录下面,他跟我说文件不存在???我从官网下载下来的原文件名是:
cuda_10.0.130_410.48_linux.run
反正就运行不成功,看了一些博客后,把.run的后缀去掉才行。
Step2:
按q退出协议阅读;
accpet接受协议,
下图附上我的选项。
好了,关键信息来了,他问你要不要把cuda装在/usr/local这里,这里要选自己的路径。
Enter Toolkit Location
[ default is /usr/local/cuda-10.0 ]:,
/home/dailh/cuda-10.0/
还有cuda samples的安装地址,改成自己的地址:
Enter CUDA Samples Location
[ default is /root ]:/home/dailh/
一路选下来,基本上就是安装成功了。
安装cudnn:
这个简单,去官网下载对应的cudnn。
比如我下载的是:
cudnn-10.0-linux-x64-v7.5.1.10.tgz
解压:
tar -xzvf cudnn-10.0-linux-x64-v7.5.1.10.tgz
解压后的操作:(解压后的文件名就叫cuda)
cp /home/dailh/cuda/include/cudnn.h /home/dailh/cuda-10.0/include/
cp /home/dailh/cuda/lib64/libcudnn* /home/dailh/cuda-10.0/lib64
chmod a+r /home/dailh/cuda-10.0/include/cudnn.h /home/dailh/cuda-10.0/lib64/libcudnn*
最后步骤:
修改.bashrc:
export PATH=$PATH:/home/dailh/cuda-10.0/bin/
export PATH=$PATH:/home/dailh/cuda-10.0/lib64/
然后再:
source .bashrc
就可以使咱们的新环境生效啦
在终端输入:
nvcc -V
检查一下是不是cuda10.0:
成功!