yolo环境搭建与pytorch中torch、torchvision、torchaudio安装

 安装anaconda:

Anaconda安装(Python) - 知乎 (zhihu.com)

2020.2对应的python版本为3.7(不推荐)

2020.11对应的python版本为3.8(推荐python3.8及以上,但是尽量不要用最新的,存在兼容问题)

注:添加环境变量时,没有user这一文件夹不影响

环境搭建:

(4条消息) win10深度学习环境搭建_江南小赣的博客-CSDN博客

cuda加速器版本选择:Release Notes :: CUDA Toolkit Documentation (nvidia.com)

cudun安装:cuDNN Archive | NVIDIA Developer

GPU版本:注意,要完整添加好环境变量(4条消息) GPU版本的PyTorch安装与环境配置_「已注销」的博客-CSDN博客_安装gpu版本的pytorch

环境搭建及扩充详细教程:(4条消息) YOLO pytorch 环境配置及运行说明 (Windows环境)_alone_5的博客-CSDN博客_yolo环境

两个链接中我觉得重要的点:

(1)“首先安装Pytorch,注意安装的pytorch要与安装的cuda的驱动版本之间相互兼容,否则不能正常工作,安装命令如下:conda install pytorch torchvision cudatoolkit=11.3

若安装速度过慢或者卡住了,可以使用清华镜像进行安装。添加清华源,输入以下命令:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/(这个不能用不知道是不是网络问题)
接下来输入python,进入python编辑器,输入import torch测试pytorch是否已经安装成功,再输入torch.cuda.is_available()查看cuda能否调用GPU”
 

“在Anaconda prompt中操作:增加配置包的速度

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
查看是否修改好通道:

conda config --show channels”

注:如果要删除自定义源更换回conda的默认源,直接删除channels即可,命令如下:
conda config --remove-key channels

pytorch中torch、torchvision、torchaudio安装

对应关系:

(4条消息) PyTorch中torch、torchvision、torchaudio版本对应关系_Liekkas Kono的博客-CSDN博客_torchaudio版本

安装教程 :

(4条消息) 如何安装torch、Pytorch、torchversion_-Han-的博客-CSDN博客_torch.version

torch系列网站下载:

https://download.pytorch.org/whl/torch_stable.html

其他一些包的安装

tqdm安装:pip install tqdm

yaml安装:pip install pyyaml

matplotlib安装:pip install matplotlib

scipy安装:pip install scipy

tensorboard安装:pip install tensorboard

seaborn安装:pip install seaborn

thop安装:pip install thop

最麻烦的是pycocotools:添加个coco数据集就行

错误的解决:

错误1:AttributeError: module 'distutils' has no attribute 'version'

AttributeError: module ‘distutils‘ has no attribute ‘version‘ 解决方案 - 知乎 (zhihu.com)

原因分析:setuptools版本问题”,版本过高导致的问题。

解决方法:pip uninstall setuptools

conda install setuptools==58.0.4

 
 

你可能感兴趣的:(安装包,深度学习,python,pytorch)