Ubuntu18.04+CUDA11.0+cuDNN+PyTorch1.7.0(GPU) |
您所在的位置:网站首页 › cudnn80安装 › Ubuntu18.04+CUDA11.0+cuDNN+PyTorch1.7.0(GPU) |
Ubuntu18.04+CUDA11.0+Cudnn+Pytorch1.7.0(GPU)
1.CUDA安装
1、系统要求 从表格可以查找系统对应的gcc与kernel版本 1)是否安装了gcc编译器,查看当前系统的gcc版本 gcc --version信息显示 2)确认系统安装了满足要求的Kernel Header # 查看系统正在运行的Kernel Header uname -r信息显示 2、 查看显卡信息,有显卡是安装cuda的前提。 lspci | grep -i nvidia信息显示 3、安装NVIDIA显卡驱动 注意:现在显卡驱动可以在CUDA安装包一起安装,但我的显卡驱动是提前装好的,所以具体安装自行查找,这里简略介绍涉及的一些信息。 显卡驱动下载可以参考该博客:https://blog.csdn.net/tanmx219/article/details/107591416 1)显卡具有推荐推荐的安装版本 ubuntu-drivers devices信息显示 2)查看安装显卡驱动后的显卡信息 nvidia-smi信息显示 4、开始安装 Table 2 来源 注:建议下载runfile(后缀为.run),deb是另一种类型的安装文件 无输出可以直接进行下一步,但有输出则代表nouveau正在加载,需要手动禁掉nouveau 禁用方法可参考该链接博客内容,但我没有亲自尝试:https://blog.51cto.com/14204744/2464180?source=dra 3)进入该文件所在文件目录,给文件赋予执行权限 sudo chmod a+x cuda_11.0.2_450.51.05_linux.run(上述命令不太懂有没有必要,但是执行了也没问题) 4)根据官网给的命令,运行run文件 sudo sh cuda_11.0.2_450.51.05_linux.run第一步:continue 往下都是yes或者all install之类的,由于是在安装之后才写了说明文档,无图不再赘述 安装完成:安装中没有选择Driver,所以是Not Selected,若安装中也没有选择CUDA Samples 11.0,那么Samples那一栏也会显示Not Selected 5)添加环境变量 打开主目录下的.bashrc文件在末尾添加这两行 export PATH=$PATH:/usr/local/cuda-11.0/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64注意:其他版本的cuda就要改一下cuda-xx.x 保存文件后,执行命令,使环境变量生效 source ~/.bashrc6)验证CUDA是否安装成功 方法一: nvcc -V信息显示 信息显示 5、卸载CUDA 参考博客:https://blog.csdn.net/qq_33200967/article/details/80689543#_59 2.cuDNN安装1)官网下载cuDNN:https://developer.nvidia.com/rdp/cudnn-archive 根据系统和CUDA版本选择安装包,8.0.3还是8.0.4影响不大,一般选择新的就可以
3)验证 cp -r /usr/src/cudnn_samples_v8/ ~/ cd ~/cudnn_samples_v8/mnistCUDNN make clean && make ./mnistCUDNN信息显示 1)官网下载地址:https://www.anaconda.com/products/individual#download 清华镜像包下载地址:https://mirror.tuna.tsinghua.edu.cn/help/anaconda/ 安装命令 sudo bash Anaconda3-2020.11-Linux-x86_64.sh安装过程图可参考该博客:Ubuntu16.04下Anaconda3的安装 3.PyTorch安装1、建议先创建一个虚拟环境 conda create -n pytorch1.7 python=3.7 # pytorch1.7 自定义的环境名称 # python=3.7 指定该环境下的python版本conda info --envs # 查找虚拟环境 conda创建环境时可能报错:NotWritableError: The current user does not have write permissions to a required path 2、启动(激活)虚拟环境 conda activate pytorch1.7信息显示 PS:退出当前虚拟环境命令 conda deactivate 查看所有的虚拟环境 conda info -e 删除虚拟环境 conda remove -n pytorch1.7 --all 3、安装 2022年3月22日更正,官网界面已无11.0选项,但其他页面有pytorch与CUDN版本对应的显示 2022.3.22添加版本链接https://pytorch.org/get-started/previous-versions/ 在同一环境下建立了新的虚拟环境安装PyTorch,使用下图灰色部分的命令,同样需要如下方步骤3) 中检查列出的安装包版本是否与黑图中一致,不为cpu版本。 1)官网链接 2)添加清华镜像 由于镜像网址时不时地有变动,最好搜索最新的清华镜像进行添加 参考1(讲得比较细) 参考2 参考3 综合了3个博客的信息,最后添加命令为 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ conda config --set show_channel_urls yes.condarc文件显示为 3)需要先查看官方涉及的下载包 命令是图3-1中的红框 conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch信息显示 关注pytorch安装包的版本 4)查看镜像给予的下载包 命令是图3-1中的黄框 conda install pytorch torchvision torchaudio cudatoolkit=11.0注意:对比两种命令涉及的安装包版本,镜像命令下的安装包容易因为网址的不全、不能提供最新的安装包,导致pytorch会配cpu版本 镜像包的版本与官方一致后,再由Proceed([y]|n)?选择y进行安装 5)验证 #查看pytorch版本 import torch print(torch.__version__)信息显示 信息显示 还可以选择pip离线安装PyTorch,不需要额外配置镜像 离线安装pytorch ubuntu18.04离线安装pytorch |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |