ubuntu下安装pytorch和TensorFlow

       最近几天都在安装这两个环境,但由于是在服务器上安装,多少还是有些不方便。根据NVIDIA-smi,安装了cuda10.2,之后安装了对应的cudnn,安装对应的pytorch。

import torch

出现了OSerror,

 查了半天,大概意思环境不匹配,所以又装了cuda10.1.还是不行。走投无路看到了一篇博客(49条消息) TensorFlow/PyTorch和cuda等版本对应关系_gyxx1998的博客-CSDN博客_pytorch和cuda对应版本

试着装一下TensorFlow,结果很顺利就安装好了。非常感谢这位博主解救了我的周末。

安装TensorFlow的可以按照这篇博文就可以。

想着pytorch应该也可这样安装,于是打开pytorch主页,发现了这句:install previous version of pytorch

打开后找到对应的版本就可以在你创建的虚拟环境中安装了。感谢强大的anaconda!!!

步骤:(1)conda create -n envs-name  python=3.8   创建虚拟环境

             (2) conda activate envs-name    激活虚拟环境

             (3)打开pytorch官网,点击install previous version of pytorch

          (4)conda install pytorch torchviudio cudatoolkit=10.2 -c pytorch    安装pytorch,顺便把它需要的其他东东都自动帮你安好,再不用自己安装cuda和cudnn了。

           (5)坐等安装

          (6)python

                   import  torch

                   ...紧张的等待,没有error

           (7)torch.cuda.is_available()

                    出现True

         就成功啦!!!

花了3天没搞定,这个方法5分钟就安好了。

希望大家都不要走弯路,浪费了时间。

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值