Pytorch配置——为深度学习YOLOv5配置环境记录

‘‘‘版本与我相同可以直接从网盘里下载部分安装包
链接：https://pan.baidu.com/s/1xRsnphEYBKDiMTaibMOuIw
提取码：0904
–来自百度网盘超级会员V2的分享’’’

1 安装Anaconda

选择合适的版本安装python（推荐3.8）电脑上下多个版本的python不冲突，可以额外下一个3.8，以下为python官网下载地址：

Python Releases for Windows | Python.org

Anaconda清华源下载地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

官网下载较慢，可以在清华大学的镜像源网站上进行下载，下载版本不要太新也不要太旧。安装中的这个界面要全部勾上

下载完成后打开红线所标应用，之后会多次用到，可以右键单击–打开文件位置–再次右键单击–显示更多选项–发送到–发送到桌面快捷方式。

2 安装Visual Studio 2019

在安装CUDA之前需要先安装VS 2019，否则之后使用时会有很多莫名其妙的报错，VS 2019的安装包在百度网盘链接里有。安装时勾选上Python开发

3 安装CUDA

打开anaconda的Anaconda Prompt输入命令:

nvidia-smi

右上角可以看到我的显卡支持的最高版本为12.3，如果不显示此页面则需要更新显卡驱动，以下为官网链接：

NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

CUDA版本不用太高，如果支持建议下载11.3版本，其他版本CUDA下载连接：

CUDA 工具包存档 |NVIDIA 开发者

点击右下角“Download”

下载完成后安装即可，安装选项中勾选“自定义（高级）”，下一个界面的勾选框也勾上，默认安装在C盘即可。

安装完成之后打开anaconda的Anaconda Prompt输入命令:

nvcc -V

显示如上内容即为安装成功。

4 创建虚拟环境

Anaconda常用命令如下

创建虚拟环境: conda create -n pytorch python=3.8#创建名为“pytorch”的环境，且Python版本为3.8.x
查看已经创建的虚拟环境:conda env list
切换/进入环境:conda activate pytorch
删除环境:conda remove -n pytorch --all (退出该环境再进行操作)
复制环境:conda create -n new_name --clone old_name
查看包:conda list
查看显卡状态及使用情况:nvidia-smi

打开anaconda的Anaconda Prompt输入命令:

conda create -n test python=3.8

输入y点击回车，环境配置完成后输入：

conda activate test

进入“test”环境

5 安装pytorch

查表确定所安装的CUDA版本对应的 PyTorch版本

PyTorch及torchvision建议离线安装，离线安装包下载地址

download.pytorch.org/whl/torch_stable.html

找到对应版本，我的CUDA版本为11.3，

cu113表示CUDA版本为11.3，torch-1.11.0表示torch版本为1.11.0，cp38表示Python版本为3.8.x(python版本建议3.8），Windows系统下下面的，Linux系统下上面的。

下载完成之后找到文件保存路径并复制，先进入文件所在的盘，比如说在E盘，则输入指令：

E:

（显示的报错是因为冒号是中文输入法的冒号，注意要用英文输入法的冒号）

进入文件所在文件夹，输入命令：

 cd 文件路径

如图已进入文件所在位置

输入命令并回车：

pip install "torch-1.11.0+cu113-cp38-cp38-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple/

Pip install 后的
"torch-1.11.0+cu113-cp38-cp38-win_amd64.whl"为离线安装包名，输入pip install torch后按Tab键可自动补全，-i后网址为清华大学镜像源网站，加上下的比较快。

以下为常用镜像源网站

阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣 http://pypi.douban.com/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

下载完成后输入”python”进入python模式

输入如下代码，查看是否成功安装

import torch
print(torch.__version__)#查看pytorch版本
print(torch.cuda.is_available())#验证GPU能不能用，能用返回Ture

显示如上结果表示安装成功。

Ctrl+z退出python模式

6 安装torchvision

查表找到pytorch对应的torchvision版本，下载完成离线安装包后安装包步骤与pytorch相同，不过在使用Tab补全时要输入到torchv才可补全。输入命令：

pip install "torchvision-0.12.0+cu113-cp38-cp38-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple/

7 安装pycharm

官网下载即可:

Download PyCharm: The Python IDE for data science and web development by JetBrains

*设置为中文：*打开pycharm，点击右上角齿轮进入设置

在Plugins中搜索Chinese，下载图示插件

下载完成后点击重启pycharm

8 Pycharm与虚拟环境的连接

重启完成后即为中文界面，在右下角打开Python解释器–添加新的解释器–添加本地解释器

点击Conda环境，提示找不到Conda文件

点击输入框右边的文件夹图标，找到Anaconda文件夹下的Scripts文件夹中的conda.exe文件

使用现有环境，点击test即可使用之前配置好的“test”虚拟环境