- 博客(4)
- 收藏
- 关注
RSS订阅
原创 windows下tensorRT7.0，cuda10.2 需要的环境安装
1.首先安装Anaconda 对应的python版本是3.7 2.安装cuda10.2 3.安装cudnn对应的cuda的版本 4.安装pytorch1.5 找到cud102 5.安装torchvision 0.6 对应cud10.2 为什么要用cud10.2： 因为可以使用tensorRT
2020-07-29 23:27:31
1193
原创 conda安装pytorch遇到的问题
1.安装显示通道，不能继续安装。 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/ 原因：只是换了清华园，没有换pytorh的源 解决方法： 继续添加源： conda confi...
2020-11-08 13:13:05
743
原创 torch的张量操作
一、torch.stack a = torch.tensor([[1,2,3]]) b = torch.tensor([[4,5,6]]) print(torch.stack((a,b),dim=0)) print(torch.stack((a,b),dim=1)) #输出: tensor([[[1, 2, 3]], [[4, 5, 6]]]) tensor([[[1, 2, 3], [4, 5, 6]]]) 二、torch.cat a = torch.
2020-09-07 21:54:06
1925
原创 python中的几种矩阵乘法运算
一、np.dot() 或者 @ 1.同线性代数中矩阵乘法的定义。np.dot(A, B)表示： 对二维矩阵，计算真正意义上的矩阵乘积。 对于一维矩阵，计算两者的内积。 a1 = np.array([[4,5,6],[1,2,3]]) a2 = np.array([[1,2],[3,4],[5,6]]) #矩阵叉乘 #print(a1.dot(a2)) # print(a1@a2) #输出：[[49 64] [22 28]] #对于一维信号 b1 = np.array([1,3,6].
2020-09-07 21:30:08
2873
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人