1. 首页 > 汽车 >

tf哪些算子调用了pytorchcuda版本?

cuda11.8找不到对应的pytorch

版本不匹配。指定CUDAToolkit版本根据表一查询到可安装的CUDAToolkit版本,384.81对应的CUDAToolkit版本为9.0。运行condainstallpytorchcudatoolkit=9.0-cpytorch即可。

tf哪些算子调用了pytorchcuda版本?tf哪些算子调用了pytorchcuda版本?


tf 哪些算子 调用了cuda

这俩用的不同的库,主要不同时读到一个项目里问题应该不大。CUDA只是把深度学习的算子编译了一下。下面还是要调用系统CUDA driver的,只是每次tf和pt切换的时候,要手动设置对应cuda版本的环境变量... 或者用conda搞个环境也行。

这里使用conda虚拟环境,每个环境采用不同的cuda版本,注意,这种方式必须保证驱动程序版本足够高,起码能够支持虚拟环境中使用的cuda版本。

例如,想要使用pytorch1.6,你可以使用conda install pytorch=1.6 cudatoolkit=10.2来在一个虚拟环境中配置,也可以使用conda install tensorflow=2.3.1 cudatoolkit=10.1 cudnn=7.6.5在另一个虚拟环境中配置Tensorflow的相关环境。

CUDA版本与GPU驱动版本问题

GPU的驱动版本决定了支持CUDA的版本,所以安装tensorflow gpu(或者pytorch gpu)版本时,版本选择顺序一般为:

tensorflow gpu(或pytorch gpu)版本---->CUDA,cuDNN版本---->GPU驱动版本

注:当然安装CUDA时可以选择一起安装驱动,此驱动与CUDA版本是适配的

在桌面上单击右键,然后打开NVIDIA控制面板

点击系统信息,然后选组件,查看支持的CUDA版本

笔记本1650显卡初始驱动

第一步,查询电脑显卡驱动信息,个人电脑显卡查询如下

步骤如下:

1首先,我们打开笔记本电脑,我们点击打开控制面板,

2接下来,我们点击打开设备管理器,

3然后,我们右击显卡,点击打开属性,

4接着,我们点击打开上方的驱动程序选项卡,

5后,我们直接点击更新驱动程序按钮,就可以安装驱动了。

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 12345678@qq.com 举报,一经查实,本站将立刻删除。

联系我们

工作日:9:30-18:30,节假日休息