Ubuntu16.04,cuda10.0下安装pytorch0.4.0(因为很多代码要求必须是pytorch040,cuda8.0或者9.0)cuda10.0下安装pytorch040,发现cuda版本pytorch版本不匹配,代码运行过程中会报错,所以为了方便便在服务器自己的个人环境下装了新的cuda9.0,因为服务器是2080ti,比较好,1080ti也挺好的,都可以用这个方法给自己的环境下安装新的cuda而不会影响大环境(local)下的cuda版本。再说明一下,2080ti只能安装cuda版本大于等于9.0的,但是1080ti可以安装cuda=8.0,博主用该方法实验过在2080ti上安装cuda9.0在1080ti上安装cuda8.0均成功,教程详细,适合新手,欢迎大家批评指正,转载注明出处!!!
方法如下:
(1)下载安装文件。首先去英伟达官网下载cuda安装包:https://developer.nvidia.com/cuda-toolkit-archive
按照系统配置选择安装包,据说用deb文件安装容易出错,所以直接下载runfile(后缀为.run)文件来安装,例如我想安装cuda9.0,且我的操作系统为Ubuntu16.04,所以我按照下图选择安装包:
下载完成之后上传到服务器上去,上传位置可以随意,然后执行:
sh cuda_9.0.176_384.81_linux.run
记住不能用sudo指令,安装在自己个人的环境下不能用sudo管理员指令。
然后一直按住回车键,知道进度条到达100%。
出现:
输入:accept
然后:不安装driver
输入:n
然后:
输入:y
注意装到自己的个人环境里!
不装到大环境里,选择n,再是y,然后最后一行default路径正确直接回车,开始安装。
然后按照提示,将这三个路径添加到.bashrc文件的最后:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/wushan/.conda/envs/py040/lib64
export PATH=$PATH:/home/wushan/.conda/envs/py040/bin
export CUDA_HOME=$CUDA_HOME:/home/wushan/.conda/envs/py040
最后验证一下:
nvcc --version
成功!