服务器安装cuda、tensorflow-gpu环境

1、安装Nvidia驱动(服务器预装了驱动,这步可不做)

1、先查看显卡 lspci | grep NVIDIA
2、再查看驱动版本 nvidia-smi
如果没有安装,使用这个命令安装 (必须禁用nouveau 链接
ubuntu安装显卡驱动的三种方法

以下这步可以不做
使用这个命令,切换更换内核 链接

2、安装cuda-9.0, cudnn-7.6.0

此时nvidia 驱动版本必须大于384.81
安装方法:Ubuntu16.04下安装cuda和cudnn的三种方法

2、安装Anaconda3,避免和vncserver冲突

  • 在 sh Anaconda.sh的时候,将安装位置改为/usr/local/anaconda3,并且在创建快捷方式那里输入no
    (即:拒绝创建环境变量到~/.bashrc下,先不要添加,选择no)

  • 安装完毕后,sudo vim /etc/profile, 在文件末尾输入

    PATH=$PATH:/root/anaconda3/bin  #路径名跟自己实际情况而定
    export PATH
    

    接着保存退出,输入 source /etc/profile 使配置生效
    - vncserver支持的是python2.7,而新安装的Anaconda3安装的是python3

参考链接:
安装 Python 相关组件导致 VNC 界面黑屏的解决办法
Liunx系统添加anaconda环境变量

接着reboot,重启

  • 使用conda create -n py36 python=3.6创建虚拟环境
  • conda activate 创建的环境
  • 使用pip install -i https://pypi.douban.com/simple 安装自己的需要的包,例如tensorflow-gpu-1.8
    参考链接:
    解决pip下载包速度慢的问题
  • 运行程序的时候记得写 python3 xxx.py 而不是python xxx.py

3、安装Pycharm,并且使用Anaconda的软件库

	sh pycharm.sh
	有时候会出现tensorflow-gpu不能使用的问题,可以将gcc\g++设置位4.8

4、服务器上传、下载文件使用的是Putty自带的pscp

打开cmd,cd到Putty路径下

  • 上传:
pscp D:\Anaconda3-2019.03-Linux-x86_64.sh  [email protected]:/root/
  • 下载:
 pscp [email protected]:/root/Anaconda3-2019.03-Linux-x86_64.sh.zip D:\

如果使用pscp上传文件时 出现FATAL ERROR: Remote side unexpectedly closed network connection
关闭cmd,重复上述操作

发布了59 篇原创文章 · 获赞 0 · 访问量 698

猜你喜欢

转载自blog.csdn.net/wuprogrammer/article/details/104736057