cuda

【记录】 编译matconvnet on ubuntu16.04 with cuda 9.0

陌路散爱 提交于 2020-11-22 16:35:48
最近需要用到matconvnet 在Ubuntu16.04下。 因为TensorFlow 1.6 支持了CUDA 9.0 所以新机器就直接装了9.0 但是编译matconvnet 时遇到了一些问题 特此记录一下。 1. Error using mex nvcc fatal : Unsupported gpu architecture 'compute_20' Solution: 这个是因为cuda 8 之后不支持compute_20 了,最低也是compute_30了。 所以需要将vl_compilenn.m中的以下代码进行修改 opts.defCudaArch = [... '-gencode=arch=compute_20,code=\"sm_20,compute_20\" '... '-gencode=arch=compute_30,code=\"sm_30,compute_30\"']; 我用的是GTX1080TI, 此处我修改成 opts.defCudaArch = [... '-gencode=arch=compute_30,code=\"sm_30,compute_30\" '... '-gencode=arch=compute_50,code=\"sm_50,compute_50\"']; 同时还需要将 matconvnet/matlab/src/config

【目标检测-框架测试】mmdetection的安装与使用

两盒软妹~` 提交于 2020-11-21 09:36:43
1、 安装 anaconda 2、 更新 gcc 到 4.9 以上 gcc -std=c++11 test.cpp 测试代码 test.cpp #include<iostream> #include <memory> using namespace std; int main(){ shared_ptr < int > p = make_shared< int >( 42 ); cout << " p = " <<*p<< endl; return 0 ; } 3、 安装 mmdetection 1 )创建 conda 虚拟环境并激活,然后安装 cpython conda create -n open-mmlab python= 3.7 - y conda activate open - mmlab conda install cython 2 )查看本机是否支持 GPU lspci | grep -i nvidia 安装 pytorch(>=1.0) # 可以直接通过 pytorch 的官网查看安装命令 3 )克隆 mmdetection 仓库 git clone https: // github.com/open-mmlab/mmdetection.git cd mmdetection # 2019 年 7 月 27 日 更新后的 mmdetection ,将 compile

Ubuntu18.04上安装CUDA_10.1(nvidia-driver)和cuDNN_7.6.5

跟風遠走 提交于 2020-11-20 16:49:27
本文是在Ubuntu18.04.5服务器上安装CUDA_10.1(nvidia-driver455)和cuDNN_7.6.5, Ubuntu 18.04.5 CUDA_10.1 (nvidia-driver455) cuDNN_7.6.5 一、 前期准备 1、查看系统版本和显卡型号 cat /etc/issue # 查看系统版本 lspci | grep -i nvidia # 查看电脑的显卡型号 2、查看系统是否安装显卡驱动 nvidia-driver nvidia-smi # 查看显卡驱动情况(此处未安装,应无信息显示) 3、显卡支持的驱动型号 ubuntu-drivers devices # 查看显卡支持的驱动型号 此处可以看到系统推荐的驱动版本是 455: 4、安装推荐版本455的显卡驱动 sudo ubuntu-drivers autoinstall # 安装默认显卡驱动(此处为455) 5、重启服务器(无其它用户使用服务器) (sudo) reboot # 立即重启 或 (sudo) shutdown -r now # 立即重启 6、查看显卡驱动是否安装成功: nvidia-smi 安装成功会显示 7、卸载已有驱动程序(需要重启才生效) $ sudo apt-get remove nvidia-* 或 $ sudo apt-get remove --purge

网址收藏

回眸只為那壹抹淺笑 提交于 2020-11-15 12:59:38
NVIDIA 中国官方博客 http://blogs.nvidia.cn/ The Official NVIDIA Blog https://blogs.nvidia.com/ NVIDIA Developer https://developer.nvidia.com/ NVIDIA Developer Blog | NVIDIA Technical Blog: for developers... https://devblogs.nvidia.com/ https://devblogs.nvidia.com/?search_theme_form=pro+tip NVIDIA Developer Documentation https://docs.nvidia.com/ https://docs.nvidia.com/cuda/index.html Jetson Download Center https://developer.nvidia.com/embedded/downloads 来源: oschina 链接: https://my.oschina.net/u/4344016/blog/3833289

ubuntu TensorRT

不羁岁月 提交于 2020-11-13 01:05:56
star@xmatrix:~ $ star@xmatrix: ~ $ star@xmatrix: ~$ lspci |grep - i vga 08 : 00.0 VGA compatible controller: ASPEED Technology, Inc. ASPEED Graphics Family (rev 30 ) star@xmatrix: ~ $ star@xmatrix: ~ $ star@xmatrix: ~ $ star@xmatrix: ~$ lspci | grep - i nvidia 02 : 00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1) star@xmatrix: ~ $ star@xmatrix: ~ $ star@xmatrix: ~$ cat /proc/driver/nvidia/ version NVRM version: NVIDIA UNIX x86_64 Kernel Module 418.67 Sat Apr 6 03 : 07 : 24 CDT 2019 GCC version: gcc version 5.4 . 0 20160609 (Ubuntu 5.4 . 0 -6ubuntu1~ 16.04 . 12 ) star@xmatrix: