ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

3090卡torch,mmcv,mmdetection环境配置 - 全宇宙最好

2021-11-20 22:33:57  阅读:245  来源: 互联网

标签:mmcv 全宇宙 torch cudnn cuda https mmdetection PATH


本文

本文堪称全宇宙最好的安装教程!(就是吹吹牛逼呢)
安装这个环境其实很简单,就三步: 显卡驱动 —— cuda —— torch, 安装过程中,注意版本对应就可以。

显卡驱动

sudo apt install nvidia-driver-460(如果说这里无法切换驱动)

cuda版本


下载链接
https://developer.nvidia.com/cuda-toolkit-archive

设置环境变量
就是在.bashrc中把这个加进去就行
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
cudnn
如果要装cudnn的话(不是必要的)
https://developer.nvidia.com/rdp/cudnn-archive
在该网页中找到适配所用cuda的cudnn包,解压copy到cuda对应的头文件和库文件路径即可。

cudnn的好处,其GPU加速作用;
加速开关的代码为:
torch.backends.cudnn.benchmark = True

torch


torch下载说明
https://pytorch.org/get-started/previous-versions/
通常,使用pip安装就可以;

mmcv 注意要素
https://mmcv.readthedocs.io/en/latest/get_started/installation.html

mmdetection
https://mmdetection.readthedocs.io/en/latest/get_started.html#installation

标签:mmcv,全宇宙,torch,cudnn,cuda,https,mmdetection,PATH
来源: https://www.cnblogs.com/yubo455/p/15582839.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有