JZTXT
  • 首页
  • Ai
  • Java
  • Python
  • Android
  • Mysql
  • JavaScript
  • Html
  • CSS

升级显卡驱动,下载cuda和cudann 12.0后,安装pytorch

发布时间 2023-11-05 18:43:54作者: 春山可

1、重新创建虚拟环境

 2、查找pytorch版本

安装Pytorch如何选择CUDA的版本,看这一篇就够了 - 知乎 (zhihu.com)

Previous PyTorch Versions | PyTorch

安装支持CUDA 12的pytorch教程 - 知乎 (zhihu.com)

 

于是:

 

 conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

 (未开net,因为昨天开着报错了,真的好慢,还怕清华源变成CPU的...)

 完成

试了一下啊,cuda可用。

    本栏目推荐文章
  • 激光雕刻机步进电机驱动方案-TMC2226
  • 哈哈哈,我就说未来要研发无人的AI潜艇嘛 —— 说啥来啥 —— AI驱动的无人潜艇
  • 数据驱动未来:公司运营数据分析大屏引领企业变革
  • selenium加载驱动的两种方式
  • 软件测试/测试开发全日制|Pyest结合json实现数据驱动测试
  • 软件测试/测试开发全日制|Pytest结合CSV实现测试的数据驱动
  • Pytorch中的交叉熵CE和均方误差MSE分别是如何计算的?
  • 快速查看显卡使用情况和占用用户
  • devicemapper 存储驱动中的 Thin Pool 的可用数据块数量不足
  • ubuntu20.04 opencv 4.2 + opencv_contrib 4.2.0 安装笔记 和 opencv 4.5.4 with cuda 安装笔记
版权声明:本网站为非赢利性站点,本网站所有内容均来源于互联网相关站点自动搜索采集信息,相关链接已经注明来源。
联系我们