因为项目要求,本文主要使用anaconda ,pycharm,安装包使用GPU版本
暂时更新到安装包,所以其实一点yolov5的内容都还莫得
conda config --remove-key channels
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple
然后创建新环境:(我这里选的mask_yolov5代表环境名,创建时,可以参考自己的项目名或创建环境的目的)
conda create -n mask_yolo5 python==3.8
激活新环境
conda activate mask_yolo5
然后用pycharm打开项目文件夹,在pycharm的右下角可以找到添加解释器:
如果项目中含有requirements.txt,说明,,,这个人是个好人(bushi
说明可以直接安装大部分包,pycharm也会有提示“是否安装requirements里的包”。但仍有几个包需要手动安装(哪里报错按哪个)
方式:
cd 项目文档路径
pip install -r requirements.txt
笔者主要在tensorflow和pytorch安装时,屡屡碰壁,所以主要说这两个包的安装
首先,大胆尝试直接安装(记得激活环境)
pip install tensorflow-gpu==2.8.0
然后我报错了,,,第一次是TypeError: Descriptors cannot not be created directly. If this call came from a _pb2。按照网上教程把protobuf降级
pip uninstall protobuf
pip install protobuf==3.19.0
然后报错AttributeError: module ‘keras.utils.generic_utils‘ has no attribute ‘populate_dict_with_module。是keras与tensorflow版本不匹配(我真的会谢
下一个匹配的版本吧,,,
版本对应网站
然后检查一下: 先进入python(就是激活环境,然后输入python)
import tensorflow as tf
print(tf.test.is_gpu_available())
输出是true就可以
pytorch需要注意的是,用conda直接从镜像源下载,下的是cpu版本(诡计多端的镜像源 )所以我们需要从官网下载whl文件,然后安装
这里要感谢这位博主!!下载whl文件教程
然后最后再测试一下就可以了
python
import torch
torch.cuda.is_available()