随笔分类 - 深度学习工具包
Ubuntu20.04安装CUDA和CUDNN
摘要:CUDA是GPU深度学习的运行库,那么cuDNN就是训练加速工具,两者要相互配合使用,所以一般机器学习需要训练引擎(tensorflow-gpu) + CUDA + cuDNN使用。想不安装cuDNN是不可以的,而且cuDNN版本要和CUDA版本相互搭配。 1、前置工作 查看显卡信息 nvidia-
官网使用conda&pip安装PyTorch命令总结(包含各版本)
摘要:原网页https://pytorch.org/get-started/previous-versions/ 因为有时访问该网站比较慢,所以本博客记录该网页内容 Installing previous versions of PyTorch We’d prefer you install the la