从0到1全流程使用 segment-anything
一、安装 anaconda
一、下载 anaconda
二、以管理员身份运行安装
1、勾选 Just Me
2、统一安装路径(后续 python 等包也安装至此目录)
3、勾选 add to path
然后安装即可。
三、修改 Anaconda 默认路径及默认缓存路径
Anaconda 默认下载的包、缓存路径为 C 盘,为了防止 C 盘内存大小被占用,可以将路径进行修改。
1、在 C:\Users\用户名 目录下,找到 .condarc 文件
若没有此文件,请先打开 Anaconda Navigator,启动后再关闭,再查看是否存在 .condarc 文件
2、修改 .condarc,修改默认路径
添加 envs_dirs 和 pkgs_dirs ,指定到想要存放的路径
3、打开 Anaconda Prompt,验证修改路径是否成功
若打开 Anaconda Prompt 后,出现 .condarc 打不开的报错问题,则请先删除 .condarc 文件,再次打开 Anaconda Navigator,然后再次修改 .condarc。
输入 conda info 命令,可以看到,package cache 和 envs directories 的首个路径就是我们设置的路径,则说明修改成功。
之后下载的安装包,则会安装到这里指定的路径下。
4、设置路径的权限,防止无权限安装、访问等
对于指定的 envs 文件夹 和 pkgs 文件夹设置用户完全访问、控制权限。
二、搭建虚拟环境
一、打开 Anaconda Prompt
二、输入 conda create -n seganything python=3.8
输入 y 确认:
segment anything 的 python 要求在 3.8及以上
三、安装对应 pytorch、torchvision 版本,与 CUDA 匹配
1、打开控制台,查看自己电脑 CUDA 版本
输入 nvidia-smi,可以看到 CUDA Version 为 12.4
那么安装的 pytorch 版本支持的 CUDA 版本就不能超过 12.4
2、pytorch 官网查找适合版本
需要注意的是 pytorch、CUDA、python、Anaconda 的四个版本都要在相互支持的范围内,且满足 segment anything 的要求:
即:
1、segment anything 要求 python >= 3.8 且 pytorch >= 1.7 且 torchvision >= 0.8
2、通过 create 命令创建虚拟环境时,设置的 python 为 3.8
3、pytorch 也有对 python、cuda 支持的对应版本要求
这里在 pytorch 官网可以点击 install previous versions of Pytorch 查找之前的版本,也可以在当前页面选择对应的版本,复制 Run this Command 对应的命令。
执行命令前,需要在 Anaconda Prompt 中先激活环境:
conda activate seganything
然后执行安装 pytorch 命令 这里选择执行安装最新版本(Stable 2.2.1):
conda install py