深度学习相关知识

目录

一.环境安装

1.cuda/cudnn安装：

2.Pytorch，torchvision安装

3. TensorRT安装C++版本：

4.onnxRuntime安装：

二.基础知识

1.Onnx、onnxTime、wts和TensorRT

2. 模型部署

3. 深度学习简介

4. Loss函数

5. 激活函数

---

一.环境安装

1.cuda/cudnn安装：

安装cuda之前首先安装vs，vs版本从低到高安装。

a) 安装cuda：首先查看显卡支持的最高CUDA的版本，以便下载对应的CUDA安装包；

cmd命令行：nvidia-smi,显示如下：最高版本为12.2，

在NVIDIA官方网站即可下载，地址为：https://developer.nvidia.com/cuda-toolkit-archive

无坑，双击exe选择安装目录即可，C盘有空间尽量C盘安装；安装完成使用nvcc -V确认安装成功

b) cudnn：cudnn版本选择对应支持的较高版本即可；非常简单，将bin,include,lib三个文件夹拷贝到cuda安装目录下对应的文件夹里面，下载地址：cuDNN Archive | NVIDIA Developer

c)多版本cuda：按照ab下载对应版本，安装cuda时只选择安装CUDA组件即可，之后在环境变变量path中将优先使用的cuda放置到最前面；

 

2.Pytorch，torchvision安装

使用conda创建虚拟环境，conda create -n py39 ，激活环境，activate py39，查看环境：conda env list

之后根据自己的环境进行安装，不要瞎搜索，根据官网的命令，地址：

https://pytorch.org/get-started/previous-versions/

选择符合自己环境的，比如1.10，按照如下命令即可。

镜像网址： Index of pytorch-wheels/cu117/ (sjtu.edu.cn)