PyTorch与TensorFlow都是深度学习框架。
环境安装
前面pycharm已经安装,而且安装了conda,这里不记录了。主要记录CUDA的安装,用于GPU加速计算,一般比CPU单独计算速度快50倍左右。
这里要求显卡是NVIDIA显卡。
下载地址
如果链接失效,可以谷歌搜索CUDA download
安装一律下一步,安装位置可以自己改一下。
安装完成后在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin下找到
说明安装成功,这是编译器。
然后在终端中测试一下
如果在终端测试无效,那么就需要配置环境变量,在里面添加上面所说的bin文件夹
PyTorch安装
官网
然后复制图里框框中的那句话,使用管理员权限运行cmd
把刚才复制的东西贴过来运行,等一会
输入y
然后开始
再在pycharm中检查一下
import torch
print(torch.__version__)
print(torch.cuda.is_available())
回归问题
梯度
梯度下降是机器学习的核心,值的跟新围绕下面的方式。
当然,再此基础上,我们还会添加其他的限定条件,例如添加学习效率,控制变化的幅度来尽量避免大幅震荡,选用不同求解器来更改趋向最优解的方式等等。
线性回归linear regression
链接
逻辑回归logistic regression
链接