ubuntu中pytorch安装及CUDA版本兼容问题记录

pytorch是目前主流的深度学习框架之一,在安装pytorch时经常会遇到版本不兼容等问题。下面对pytorch安装进行一些记录和分享,如有不准确的地方,欢迎评论区指出。

1 相关概念

  • 显卡驱动:显卡驱动的版本决定了我们电脑中安装的cuda版本的上限,因此,一般建议显卡驱动的版本安装的高一些,在linux终端中通过下面的命令可以查看:
  • watch -n 1 nvidia-smi
    查看显卡状态以及最高支持的cuda版本
  • cuda和cudnn:安装完显卡驱动之后就需要安装cuda和cudnn,cuda和cudnn是存在版本对应的,可以在ubuntu中安装多个版本的cuda,使用软链接进行切换,也可以在配置文件.bashrc中修改cuda路径进行切换,但是需要注意的是所安装的cuda版本不能超过显卡驱动所支持的cuda最高版本。

2 pytorch安装

pytorch的安装直接去pytorch官网查找对应的版本进行安装即可,可以安装cpu和gpu两个版本,通常都是安装gpu版本。
安装pytorch的命令有pipconda两种方式,那pip和conda这两种方式有什么区别呢?实际上,选择pip安装,就意味着我们安装的pytorch是依赖于电脑本地的cuda的,也就是上面概念中提到的,而在虚拟环境中是不会安装对应的cuda的。而conda命令安装pytorch就会在虚拟环境中安装cuda,即命令中的cudatoolkit,这意味着安装的pytorch是依赖虚拟环境中的cuda的,而不会去使用本地的cuda。对于较老的版本,conda命令安装只会安装对应的cuda,而不会安装cudnn,这就需要我们手动安装cudnn,conda install cudnn==对应版本。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值