配置环境
Ubuntu16.04系统安装好之后,研究强化学习,进行强化学习训练,跑代码,需要对我们电脑进行强化学习环境配置!
1、安装显卡驱动
(1)禁用nouveau。
打开终端输入:sudo gedit /etc/modprobe.d/blacklist.conf
在blacklist.conf文件末尾加上这两行,并保存:
blacklist nouveau
options nouveau modeset=0
然后执行命令:nvidiancascosdc
继续在终端输入:sudo update-initramfs -u //应用更改
重启系统,验证是否禁用nouveau:lsmod | grep nouveau
没有信息返回说明已禁用
(2)禁止连接WiFi
把WiFi、网卡关闭,要不然到后面进入tty1模式,tty不断出现
iwlwifi 0000:00:14.3: Unhandled alg: 0x707
这种东西,会导致你无法输入内容。
(3)下载驱动文件并指令安装。
参考链接:点击此处
2、cuda(10.0版本的)
参考链接:点击此处
测试是否成功:
打开终端,输入:nvcc -V
有输出cuda的版本信息就说明安装成功了。
3、安装cudnn以及测试是否安装成功(7.4版本的)
参考链接:点击此处
4、安装TensorFlow-gpu(1.14版本的)
将下载好的该文件放在下载的目录下,如我放在~/Downloads文件夹中,则使用以下命令安装:
cd ~/Downloads
pip install tensorflow_gpu-1.14.0-cp35-cp35m-manylinux1_x86_64.whl
5、结束
终于结束了!安装好之后,打开pycharm,运行试试!
感谢大家的阅读!