- 博客(4)
- 收藏
- 关注
原创 windows下tensorRT7.0,cuda10.2 需要的环境安装
1.首先安装Anaconda 对应的python版本是3.72.安装cuda10.23.安装cudnn对应的cuda的版本4.安装pytorch1.5 找到cud1025.安装torchvision 0.6 对应cud10.2为什么要用cud10.2: 因为可以使用tensorRT
2020-07-29 23:27:31 1126
原创 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 655
原创 torch的张量操作
一、torch.stacka = 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.cata = torch.
2020-09-07 21:54:06 1819
原创 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 2788
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人