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

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

​ 前言:在第一次配置DIM项目运行环境的失败的背景下进行第二次环境的配置安装。核心是解决pytorch的安装,重点要注意cuda和cudnn的版本兼容问题,难点是判断出正确的安装源。概括主要做了以下几件事:第一、卸载cuda和cudnn,重新安装cuda10.0.13和cudnn7.5.1。第二、安装anaconda,用anaconda创建虚拟环境。第三、修改conda默认工具包的获取通道。第四、用conda安装指定版本的pytorch和附属依赖包。最后说一点,确信之前的问题的来源是清华镜像源的pytorch包有误。详情请看下文。

4.1卸载cuda和cudnn,重新安装cuda10.0.13和cudnn7.5.1

​ 自行百度,重装看我上一篇文章:https://blog.csdn.net/my_name_is_learn/article/details/104680861
卸载看这篇文章:https://blog.csdn.net/qq_33200967/article/details/80689543

4.2安装anaconda,用anaconda创建虚拟环境。

4.2.1安装anaconda

​ 自行百度,很简单,下载anaconda的.run文件之后运行即可。如

sudo sh xxxxxxxx.run

4.2.2用anaconda创建虚拟环境

​ 在目录~/anaconda/envs/下运行下面的terminal命令建立test1虚拟环境,python版本为3.6

conda create -n test1 python=3.6

4.3修改conda默认工具包的获取通道

​ 其实本质就是修改conda所需源。具体步骤如下:

一、在终端中运行以下命令修改镜像源,terminal命令:

# 使用清华conda镜像
conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

# 使用科大conda镜像
conda config --prepend channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/

二、查看镜像源是否安装成功

conda config --set show_channel_urls yes

会生成一个~/.condarc文件,运行cat命令查看文件内容

cat ~/.condarc

显示内容为,说明修改安装成功。

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true

4.4用conda安装指定版本的pytorch和附属依赖包

​ 进入虚拟环境

进入虚拟环境命令
conda activate test1
退出虚拟环境命令
conda deactivate

​ 安装1.1.0pytorch和0.3.0torchvision,有一事不明,这个语句表示需要安装一个包,名称叫做cudnn-7.6.5,可是我环境的cudnn确实是7.5.1,不知道为啥训练就通过了,疑问先留在这里,后续有时间再去解决。

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0

​ 安装其它所需依赖包,当运行python train.py之后会提醒缺一些工具包,因个人电脑的不同而不同,缺什么安装什么即可。例如缺的是numpy,terminal命令:替换工具包名字即可。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy

换工具包名字即可。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy
发布了45 篇原创文章 · 获赞 24 · 访问量 3408

猜你喜欢

转载自blog.csdn.net/my_name_is_learn/article/details/104754639