Pytorch
文章平均质量分 89
墨柳烟
这个作者很懒,什么都没留下…
展开
-
LSTM案例airline-passengers全面解析与优化
前言刚开始学习LSTM的一些理解torch版本print(torch.__version__)1.10.2原数据下载https://raw.githubusercontent.com/jbrownlee/Datasets/master/airline-passengers.csv代码步步解析加载包们import numpy as npimport matplotlib.pyplot as pltimport pandas as pdimport torchimport tor原创 2022-03-25 19:27:35 · 3005 阅读 · 0 评论 -
pytorch-gpu安装注意事项
1. CUDA与cuDNN安装1.1 GPU驱动版本检查win+R输入cmd,回车打开。输入nvidia-smi查看当前驱动的版本号,版本数值可通过更新驱动增加。1.2 CUDA版本选择按照驱动版本与表中对应的CUDA版本选择合适的版本号。https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html如我的驱动版本为511.65,我之后安装了CUDA 10.2CUDA各版本下载地址:https://developer.nv原创 2022-02-27 10:27:18 · 707 阅读 · 0 评论