ubuntu16.04 GPU 环境 常用命令

pip安装(tensorflow-gpu为例)

pip  install tensorflow-gpu

conda安装

conda  install tensorflow-gpu

pip3安装(指定版本号只需在命令末尾添加==1.12.0版本号)

pip3  install tensorflow-gpu==1.12.0

使用清华镜像下载

pip install tensorflow-gpu==1.10 -i https://pypi.tuna.tsinghua.edu.cn/simple

指定目录安装

pip install -t D:\ProgramData\Anaconda3\Lib\site-packages torch-1.0.1-cp36-cp36m-win_amd64.whl

卸载安装(pip\pip3只需将conda换成pip\pip3)

conda  uninstall tensorflow-gpu

创建虚拟环境(conda为例)

conda create -n py36 python=3.6  #py36虚拟环境的名字  python=3.6  python版本

删除虚拟环境

conda remove -n py36 --all

激活虚拟环境

source activate py36

退出虚拟环境

source deactivate py36

查看所有创建的虚拟环境

conda env list

nohup送入后台运行

nohup python train.py >nohup 2>&1 &      #train.py运行的文件  nohup生成的日志文件

CUDA指定GPU(在要运行的代码文件里添加,python为例)

import os      # 使用第一张与第三张GPU卡
os.environ["CUDA_VISIBLE_DEVICES"] = "1,3"

查看GPU使用情况 

watch -n -9 nvidia-smi

CUDA指定GPU(在命令行输入时添加,python为例)查看GPU

export CUDA_VISIBLE_DEVICES=0 python test.py	# 表示运行test.py文件时,使用编号为0的GPU卡
export CUDA_VISIBLE_DEVICES=0,2 python test.py	# 表示运行test.py文件时,使用编号为0和2的GPU卡

查看进程号

ps -fe | grep start

根据进程号杀死进程

kill -9 进程号

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Ubuntu 16.04中安装Docker并使用GPU时出现问题,首先需要确保你的系统配置符合Docker和Caffe GPU的要求。以下是一些可能的解决方案: 1. 确保你的系统支持NVIDIA GPU:首先,检查你的计算机上是否安装了NVIDIA显卡驱动程序。可以使用以下命令检查驱动程序是否已成功安装并正在运行: ``` nvidia-smi ``` 如果该命令能够成功启动,并显示NVIDIA显卡的信息,则说明驱动程序已成功安装。 2. 安装NVIDIA Docker:Docker在GPU支持方面需要使用NVIDIA Docker插件。你可以按照NVIDIA Docker官方文档中的步骤安装,包括添加apt存储库、安装软件包和配置插件等。 3. 配置NVIDIA运行时:在使用Docker时,使用NVIDIA运行时可以确保Docker能够正确地使用GPU。你可以通过编辑`/etc/docker/daemon.json`文件来配置NVIDIA运行时,确保该文件中包含以下内容: ``` { "runtimes": { "nvidia": { "path": "nvidia-container-runtime", "runtimeArgs": [] } }, "default-runtime": "nvidia" } ``` 保存并退出该文件,然后重新启动Docker服务: ``` sudo systemctl restart docker ``` 4. 运行Docker容器使用GPU:在运行Caffe容器时,确保你在`docker run`命令中使用了`--runtime=nvidia`选项,以确保容器能够使用GPU。例如: ``` docker run --runtime=nvidia caffe-gpu-image ``` 这将使用NVIDIA运行时在容器中启动Caffe,并允许它使用GPU加速。 通过执行以上步骤,你应该能够在Ubuntu 16.04上安装Docker并成功使用GPU运行Caffe。如果问题仍然存在,请检查是否遵循了正确的安装步骤,并查看系统日志以获取更多信息。如果问题仍然无法解决,你可以搜索相关的错误信息或向社区等其他资源寻求帮助。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值