Pytorch学习笔记 - ubuntu16.04 Pytorch安装

本文是PyTorch学习笔记,详细记录了在Ubuntu16.04上安装PyTorch和CUDA的过程,包括禁用nouveau、安装NVIDIA驱动、CUDA工具包及配置环境变量,最后通过pip安装PyTorch并验证安装成功。
摘要由CSDN通过智能技术生成

Pytorch学习笔记 - Pytorch安装

前言

最近在学习Pytorch,希望通过写博客的方式,一方面记录下自己在学习过程中遇到的坑,方便自己以后查阅,另一方面也希望能够将自己的学习笔记分享出来,给遇到类似问题的人一些帮助。


开发环境准备

本机环境:Ubuntu16.04 / GTX 1660 Ti
在下载相应的安装包之前,到torch官方网站查询支持的cuda版本,如下图,选择安装cuda11.1
(注:根据在GTX 3060 Ti显卡上的测试,30系的显卡貌似不支持cuda10系,所以我在另一台3060 Ti的电脑上安装的是cuda11.1)
在这里插入图片描述
确定好了cuda的版本之后,再到CUDA Toolkit Documentation去查询适配cuda版本的显卡驱动版本

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值