ubuntu18+cuda10.2+cudnn+7.6.5+pytorch1.1.0+python3.6环境的配置安装

最新推荐文章于 2025-10-25 17:12:25 发布

原创

最新推荐文章于 2025-10-25 17:12:25 发布 · 2.7k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#cuda #ubuntu #pytorch

本文详细介绍了在Ubuntu系统中配置CUDA10.2、cuDNN7.6.5、PyTorch1.1.0及Python3.6的全过程，包括安装步骤、环境变量设置与测试方法，适用于deep-image-matting项目。

deep-image-matting项目运行环境配置说明

前言：本文主要说明如何在ubuntu系统环境下配置cuda10.2+cudnn7.6.5+pytorch1.1.0+python3.6的运行环境。本文有个假设前提就是你的ubuntu已经安装好了最新的显卡驱动并且正常运行。一共有5个大步骤，接下来依次记录说明。借鉴文章：https://www.cnblogs.com/booturbo/p/11834661.html安装cuda+cudnn和https://blog.youkuaiyun.com/JohnsonSmile/article/details/89597095创建虚拟环境+安装pytorch

5.1安装cuda10.2

5.1.1下载cuda10.2

cuda的下载链接如下，请根据自己实际所需版本进行下载

https://developer.nvidia.com/cuda-downloads

5.1.2安装cuda10.2

第一步，推荐下载‘runfile(local)’文件，理由是安装过程比较简单易懂，语句如下。在terminal运行这句命令然后回车，因为cuda自带默认会安装显卡驱动程序，这会与你之前已经安装好的显卡驱动产生冲突，所以回车之后你需要把安装显卡驱动的“钩”取消掉。

sudo sh xxxxxxxxx.run

第二步，在安装完毕之后需要编辑环境变量。操作如下：

# 打开环境配置文件，terminal命令如下：
sudo gedit ~/.bashrc
#然后在文件最后添加下面3行，保存
export CUDA_HOME=/usr/local/cuda

export PATH=$PATH:$CUDA_HOME/bin

export LD_LIBRARY_PA