使用 MMDetection 实现 Pascal VOC 数据集的目标检测项目练习(四) annaconda和pytorch

时间:2024-11-01 07:56:29

anaconda

一、下载

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

在清华镜像网页选择版本,这里选的是:

下载得到文件:Anaconda3-2022.10-Linux-x86_64.sh。将文件拷贝到ubuntu的Downloads文件夹。

二、安装

ubuntu终端中输入:

cd ~/Downloads
bash Anaconda3-2022.10-Linux-x86_64.sh

开始安装。

输入yes,不断回车,再输入两次yes,就安装好了。安装的默认位置是用户下的anaconda3文件夹。这里注意,最好是一个一个回车,不要按着回车不放,否则会错过第二个运行conda init的yes,就需要自己配置路径。

三、配置

如果不小心错过了conda init,那么安装完之后输入conda list就会显示命令未找到。那么就需要收东配置:

终端输入nano ~/.bashrc,打开文本编辑器

nano ~/.bashrc

在文本的末尾加入:

export PATH="/home/<用户名>/anaconda3/bin:$PATH"

再输入:

source ~/.bashrc

关闭后再次打开终端,输入:

conda list

就会显示conda 已安装的包:

 anaconda安装完成。

输入:

conda create -n torch python=3.9

 

 输入:y,就可以新建一个名为torch的环境。

pytorch 

一、下载

Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

本机的python=3.9,cuda=12.1,cudnn=8.92

在上面的网页,找到符合版本要求的pytorch,下载:

得到一个文件:pytorch-2.1.0-py3.9_cuda12.1_cudnn8.9.2_0.tar.bz2 

 二、安装

在Downloads文件夹下打开ubuntu控制台,输入:

conda activate torch

torch是之前创建的conda虚拟环境的名称。

输入:conda install pytorch-2.1.0-py3.9_cuda12.1_cudnn8.9.2_0.tar.bz2,系统会自动安装。

三、验证

输入:

conda activate torch
python
import torch

 激活虚拟环境,运行python,导入pytorch。

出现了下面的错误:

 输入:

conda install mkl mkl-include

会下载和安装缺少的mkl依赖包。

注意,上面的安装会把pytorch换成CPU版的,如果需要使用GPU,需要将pytorch卸载后重装。

在torch虚拟环境中输入:

pip uninstall torch

和:

cd ~/Downloads
conda install pytorch-2.1.0-py3.9_cuda12.1_cudnn8.9.2_0.tar.bz

 安装完成后,再次验证:

conda activate torch
python
import torch
print(torch.cuda._isavailable())

输出:True

pytorch安装完成。