ubuntu16.04下2080Ti显卡配置cuda10.0+cudnn7.4.2 + tensorflow 1.13.1 + pytorch 1.0.0 + pytorch 1.0.0
友情提醒,没有一个环境配置只看一个文档就能解决的,所以还是需要多百度,多谷歌。。。
- 下载相关文件
-
下载cuda,cuda_10.0.130_410.48_linux.run
-
下载cudnn,cudnn-10.0-linux-x64-v7.4.2.24.tgz
- 下载TensorRT
- 下载pytorch,(链接: https://pan.baidu.com/s/1p-8gR8xGWYMj4FBCW1uPzQ 提取码: 6666),其他版本似乎有问题
- 安装cuda
- 如果之前安装过cuda需要先删除,运行
sudo apt-get purge nvidia-cuda*
- 将默认的nouveau显示驱动加入黑名单
sudo vim /etc/modprobe.d/blacklist.conf
blacklist nouveau
options nouveau modeset=0
- 离开图形界面,进入命令行模式,
Ctrl+Alt+F1进入命令行模式,Ctr+Alt+F7表示返回图形操作界面
- 禁用X-server
sudo service lightdm stop
- 更新修改的黑名单
sudo update-initramfs -u
- 安装cuda
cd ~/Downloads
sudo ./cuda_10.0.130_410.48_linux.run
因为之前没有安装显卡驱动,所以这里要安装显卡的图形驱动【省去别的地方找显卡驱动】
- 安装完成后如果报安装不完整,需要安装cuda driver,则需要根据提示进行安装
- 修改配置文件
sudo vim /etc/profile
# 增加下面几行
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-10.0:$CUDA_HOME
- source一下,然后输入以下命令检测以下版本
nvcc -V
- 重启
- 配置cudnn
解压cudnn-10.0-linux-x64-v7.4.2.24.tgz,解压后得到文件夹cuda,当前路径仍然为~/Downloads,进行相关文件的配置
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
- pip换源
vim ~/.pip.conf
# 添加下面几行
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=mirrors.aliyun.com
- 安装tensorflow1.13.1
pip install tensorflow-gpu==1.13.1
- 安装pytorch
pip install ~/Downloads/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
- 参考
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)