Ubuntu 16.04 LTS 在anaconda环境下安装pytorch和tensorflow gpu,cuda10.0

48 篇文章 0 订阅
26 篇文章 1 订阅

准备

最新的Anaconda:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh
Anaconda3-5.3.1-Linux-x86_64.sh
下载最新的cuda和cudnn:
cuda_10.0.130_410.48_linux.run
libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb
注意:cuda_10.0以后cudnn就可以自动安装,无需解压复制到对应文件夹

更新笔记本的显卡驱动到较新的,最好是倒数第二新的版本

大家最好不要直接在官网下,官网可能没有你对应系统的驱动,比如只有linux64的,很有可能你的电脑用不了。
在这里插入图片描述直接使用上述方法更新,如果很慢,使用如下办法:

sudo add-apt-repository ppa:graphics-drivers/ppa 
sudo apt-get update
sudo apt-get install nvidia-415

这个也很慢,但是至少可以看见进度。

安装驱动容易出问题,可能会桌面没有了,记住console的一些操作

# 进入tty
ctrl + alt + f1
# 进入桌面
ctrl + alt + f
# 关闭tty的打印
dmesg -n 1
# 运行图像桌面和关闭
sudo service lightdm start  
sudo service lightdm stop  

安装cuda和cudnn:

安装cuda,安装向导操作,不要更新驱动:

# 安装cuda
chmod 777 cuda_10.0.130_410.48_linux.run
cuda_10.0.130_410.48_linux.run

安装cudnn

# 安装cudnn
dpkg -i libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb

安装conda:

一直下一步

# 安装conda
chmod 777 Anaconda3-5.3.1-Linux-x86_64.sh
sh Anaconda3-5.3.1-Linux-x86_64.sh

安装tensorflow-gpu:

可能报错,报什么就用pip更新什么

conda create -n tensorflow-gpu pip python=3.5
conda activate tensorflow-gpu
pip install tensorflow-gpu

安装pytorch:

可能报错,报什么就用pip更新什么

conda create -n pytorch36 pip python=3.6
conda activate pytorch36
pip install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
pip install torchvision

安装jupyter notebook:

在每一个虚拟环境里运行:

python -m ipykernel install --user --name drlnd --display-name "drlnd"

使用jupyter notebook可以选择kernel
在这里插入图片描述

安装spyder:

base 安装spyder,切换使用虚拟环境的kernel:

在每个虚拟环境里运行:

python install spyder-kernels==0.*

python -c "import sys; print(sys.executable)"

# 将上述path添加到interpreter

不行,因为新的python就没有spyder-kernels=0.x的版本。

每个虚拟环境均安装一个安装spyder:

这个可行,前提是卸载base的spyder,否则会一直跳到base的spyder,且无法切换kernel

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
1. 首先,您需要下载CUDA 10.的安装包。您可以从NVIDIA官网下载,也可以使用以下命令从终端下载: ``` wget https://developer.nvidia.com/compute/cuda/10./Prod/local_installers/cuda_10..130_410.48_linux ``` 2. 下载完成后,您需要安装依赖项。使用以下命令安装: ``` sudo apt-get install build-essential dkms freeglut3 freeglut3-dev libxi-dev libxmu-dev ``` 3. 然后,您需要禁用Nouveau驱动程序。使用以下命令: ``` sudo nano /etc/modprobe.d/blacklist-nouveau.conf ``` 在文件中添加以下内容: ``` blacklist nouveau options nouveau modeset= ``` 保存并关闭文件。然后,使用以下命令更新initramfs: ``` sudo update-initramfs -u ``` 4. 重启计算机并进入命令行模式。使用以下命令停止图形界面: ``` sudo service lightdm stop ``` 5. 进入下载的CUDA 10.安装包所在的目录,并使用以下命令运行安装程序: ``` sudo sh cuda_10..130_410.48_linux.run ``` 安装程序将提示您接受许可协议和选择安装选项。按照提示进行操作。 6. 安装完成后,您需要将CUDA添加到系统路径中。使用以下命令: ``` echo 'export PATH=/usr/local/cuda-10./bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10./lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc ``` 7. 最后,您可以使用以下命令验证CUDA是否正确安装: ``` nvcc -V ``` 如果安装成功,您将看到CUDA版本信息。 祝您成功安装CUDA 10.!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值