Pytorch学习与应用
Pytorch学习与应用
AI小学僧
这个作者很懒,什么都没留下…
展开
-
Pytorch-GPU环境搭建(win11+RTX3060+CUDA11.5+CuDnn8.3.2)
软件包搭配:win11 + RTX3060 + CUDA=11.5 + CuDnn=8.3.2 + python=3.10.14 + pytorch=1.12.1。结果显示,Python版本不合适(本版本的Pytorch,需要python=3.10-3.11版本之间)注意:30系列显卡只支持CUDA11.0及以上版本加速,因此本文CUDA选择v11.5。3. 测试pytorch是否安装成功。重新安装其他版本的python。在Pycharm中应用此环境。证明通过GPU计算实现加速。原创 2024-04-26 16:42:54 · 215 阅读 · 2 评论 -
2.1 Tensor张量
② 根据ndarray进行创建(torch.from_numpy())注意:生成的张量与numpy数组共用内存,改变其中一个另一个也跟着改变。① 直接进行创建(torch.tensor())结果表示:t3中的元素随着t0中的元素改变了。原创 2024-01-15 15:05:23 · 379 阅读 · 1 评论 -
1.2pytorch的安装与检验
证明pytorch已成功安装,并且可以调用GPU进行加速。1. 检验pytorch是否成功安装。原创 2024-01-15 14:01:38 · 388 阅读 · 1 评论