相信很多学深度学习的同学,都为环境苦恼过,不想别人乱动自己辛苦配好的环境,想要一个单独属于自己的存储空间,下面是我用过用过最简单、安全、整洁的学习环境搭建方法,希望可以帮到你。
#Step1.home下用户创建配置(这一步是在root环境下)
首先在home文件夹下创建用户、设置密码
#1创建用户
sudo adduser yourname
sudo passwd yourname
#2给yourname添加root权限
打开visudo
添加 :
yourname ALL =(ALL:ALL) ALL
#Step2.cuda cudnn环境配置(此后几步都是在你的用户下)
这一步的前提是root环境下已经安装好cuda cudnn
打开cuda安装目录下.bashrc文件,将cuda的环境变量拷贝到你用户下的.bashrc文件中:
export PATH="/home/xxx/cuda9.0/bin:$PATH"
export LD_LIBRARY_PATH="/home/xxx/cuda9.0/lib64:$LD_LIBRARY_PATH"
#Step3.anconda 安装
在你的用户底下运行以下代码:
bash "/home/xxx/software/Anaconda3-5.1.0-Linux-x86_64.sh"
#Step4.pytorch 安装
或者直接运行以下安装代码:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
测试代码:import torch
#Step5.tensorflow 安装
安装tensorflow用下面的代码:
pip install tensorflow-gpu
安装好之后用下列代码测试一下:
import tensorflow as tf
tf.test.is_gpu_available()