查找合适的GPU基础镜像
安装支持python运行环境
安装tensorflow
# 腾讯云安装
pip install tensorflow-gpu==1.14.0 -i https://mirrors.cloud.tencent.com/pypi/simple/
# 删除安装 tensorflow==1.14.0
安装驱动
conda install cudatoolkit-10.1.243-h6bb024c_0.tar.bz2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
conda install cudnn-7.6.5-cuda10.1_0.tar.bz2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
上面安装过程较慢,可以下载下来直接安装
清华下载地址
北外下载地址
# 安装
conda install --use-local cudatoolkit-10.1.243-h6bb024c_0.tar.bz2
conda install --use-local cudnn-7.6.5-cuda10.1_0.tar.bz2
测试能否使用
测试代码链接
服务器前提要求:
- 已安装nvidia
- 已安装docker
如有帮助,请您三连,多谢