深度学习 环境配置

在安装好annaconda和英伟达驱动之后,开始进行环境创建及第三方插件安装,以下均在命令行进行。

1.我这里以SSD为例进行环境创建(注意一份代码一个环境,避免因包版本问题的环境污染!)

conda create -n SSD python=3.8

2.激活创建环境

conda activate SSD

3.查看自己CUDA安装的版本,可以看到是11.6

nvidia-smi

深度学习 环境配置_第1张图片

4.根据CUDA Version: 11.6去PyTorch官网下载对应版本的pytorch:PyTorch

 深度学习 环境配置_第2张图片

5.将红色内容在命令行进行输入(确定所激活的环境为SSD)

6.安装完毕后,可以检查下环境的包与代码中requirements之间的区别,在requirements文件中删去已经安装的包,避免重复安装。

深度学习 环境配置_第3张图片

 这里已经含有torchvision故不用再次安装。

7.指定requirements所在路径后,安装第三方库。

cd C:\Users\34790\Desktop\SSD-master -pytorch\SSD-master\
pip install -r requirements.txt

8.环境配置成功了。

你可能感兴趣的:(深度学习,人工智能)