ubantu 配置 gpu
时间: 2024-02-07 09:02:46 浏览: 68
Ubuntu 更新GPU显卡驱动方法.docx
要在Ubuntu上配置GPU版TensorFlow,你需要执行以下步骤:
1. 下载并安装Anaconda。可以使用以下命令下载Anaconda安装脚本:
```
wget https://repo.continuum.io/archive/Anaconda3-2019.10-Linux-x86_64.sh
```
然后运行该脚本进行安装。
2. 下载并安装cuDNN。如果Anaconda没有自动安装cuDNN,可以使用以下命令安装:
```
conda install cudnn=7.6.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
```
3. 安装GPU版TensorFlow。可以使用以下命令安装TensorFlow-GPU:
```
pip install -i https://pypi.douban.com/simple/ --upgrade tensorflow-gpu==1.14.0
```
这样,你就成功配置了Ubuntu上的GPU版TensorFlow。请确保你的显卡支持CUDA和cuDNN,并且已经正确安装了NVIDIA驱动程序。
阅读全文