tensorflow-gpu对应的CUDA和cuDNN版本在conda环境里面怎么下载
时间: 2024-06-09 15:05:11 浏览: 188
cuda各版本可以从官网下载,无需登录!cuDNN需要登录才能下载,故分享出来
5星 · 资源好评率100%
在conda环境中下载tensorflow-gpu时,会自动下载与其兼容的CUDA和cuDNN版本。但是,如果需要手动下载CUDA和cuDNN版本,则可以按照以下步骤进行操作:
1. 打开NVIDIA官网并登录账户:https://developer.nvidia.com/
2. 进入CUDA Toolkit Archive页面:https://developer.nvidia.com/cuda-toolkit-archive
3. 在页面中找到与tensorflow-gpu版本兼容的CUDA Toolkit版本,并下载安装包。
4. 进入cuDNN Archive页面:https://developer.nvidia.com/rdp/cudnn-archive
5. 在页面中找到与CUDA版本兼容的cuDNN版本,并下载对应的cuDNN安装包。
6. 将下载的CUDA和cuDNN安装包放在一个文件夹中,并在conda环境中使用以下命令进行安装:
```
conda install tensorflow-gpu=X.X.X cudatoolkit=X.X.X cuDNN=X.X.X
```
其中,`X.X.X`代表对应的版本号。
阅读全文