ubuntu共存安装cudn10.0和cudn10.2并随时切换,安装cudnn7.6.5+pytorch1.7.0

本文详细介绍了如何在Ubuntu20.04系统中,保持CUDA10.0不变的情况下,安装CUDA10.2,以及如何离线安装PyTorch1.7.0和torchvision0.8.1,特别关注了环境变量配置和依赖库的安装。
摘要由CSDN通过智能技术生成
(1) 环境配置:
  • 先说现有环境:
    ubuntu20.04/2080ti/
    nvidia Driver version:显卡驱动:440.82
    cuda 10.0 + cudnn + 7.4.5

因为需要安装pytorch-gpu,查看官网在线安装(太慢,不推荐):
https://pytorch.org/get-started/locally/
在这里插入图片描述
并没有符合cuda10.0的在线版本。尝试cuda9.2,但是太慢了翻墙都慢。
所以直接更新环境:

  • 待安装环境:
    cuda 10.2 + cudnn 7.6.5 + pytorch 1.7.0

当然我们不希望改动显卡驱动,也需要保留原来的cuda10.0环境不动:

(2) cuda10.0 + cuda10.2共存安装

nvidia官网下载cuda10.2

https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

在这里插入图片描述

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

可以看到下载的文件版本是:cuda_10.2.89_440.33.01_linux.run
表示该cuda版本:cuda_10.2.89
对应显卡驱动的最低版本:440.33.01 (我的驱动版本是440.82,因此可以使用)

  • 共存安装cuda10.2
    此处参考:https://blog.csdn.net/qq_30683995/article/details/82745246
    sudo sh cuda_10.2.89_440.33.01_linux.run安装cuda10.2:
Do you accept the above EULA? (accept/decline/quit):
accept

在这里插入图片描述
已安装显卡驱动,不选Driver,不安装示例Sample:
在这里插入图片描述
然后,如果我们只是安装这个版本,暂时不使用。选择option, Toolkit option,取消cuda软链接:
在这里插入图片描述
done -> 返回上层进行安装。速度很快,因为只需安装对应的库。
安装结束,进行配置:

  1. 环境变量
Tensorflow官方安装历程要求注意的是:配置PATH和LD_LIBRARY_PATH和CUDA_HOME环境变量.
 
vim ~/.bashrc #修改配置文件(如果你用的是zsh,则需要修改 ~/.zshrc文件)
 
#在文件结尾处添加,因为我们安装时没有修改安装路径,默认路径在/usr/local/cuda-10.2
#将原来的cuda-10.0版本环境变量修改即可:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
  1. 修改软链接
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda
(3) 下载离线安装pytorch+torchversion

下载时请注意python cuda cudnn的版本

  1. 去conda官网https://anaconda.org/pytorch/pytorch/files下载pytorch:
    我下载的版本:pytorch-1.7.0-py3.6_cuda10.2.89_cudnn7.6.5_0.tar.bz2
  2. 去conda官网下载torchvision:https://anaconda.org/pytorch/torchvision/files
    我下载的版本:torchvision-0.8.1-py36_cu102.tar.bz2
(4)离线安装

最好在虚拟环境中操作:
我的:conda create -n pytorchpro python=3.7

conda install --offline pytorch-1.3.1-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
conda install --offline torchvision-0.4.2-py37_cu101.tar.bz2

安装完成后,还需要通过pytorch官网的在线安装必要的依赖:
但不需要 -c pytorch

conda install pytorch torchvision cudatoolkit=10.2

然后就可以import torch使用了

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值