小编Oli*_*ver的帖子

使用 CUDA 11.2 和 cuDNN 8.1-8.2 设置 TensorFlow-GPU Conda 环境(CUDA 460 驱动程序

我正在尝试为tensorflow-GPU 创建一个conda 环境。我有一个 GeForce RTX 3080,运行 Ubuntu 21.04,安装了 CUDA 460 驱动程序(禁用安全启动)。我尝试使用conda安装cudatoolkit,但是使用conda可用的最新版本是11.0,它最多只支持CUDA驱动程序450。我在网上找不到任何关于如何将新的CUDA版本安装到conda环境中的信息,只能在全局上找到使用 或 .run 文件的环境sudo(我想远离它)。我假设我使用pip install cudatoolkit=11.2或类似的东西,但我不确定。

任何帮助将不胜感激,因为我是 Linux 新手,而且我还不是终端忍者。

谢谢

nvidia cuda conda tensorflow gpu-driver

3
推荐指数
1
解决办法
8167
查看次数

标签 统计

conda ×1

cuda ×1

gpu-driver ×1

nvidia ×1

tensorflow ×1