欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu16.04 通过anaconda建立虚拟环境,安装cuda9.0,cudnn7.1.2

程序员文章站 2022-05-27 13:26:10
...

一、anaconda建立虚拟环境及conda操作

//env_name代表你想要建立的环境名字,-n表示名字
conda create -n env_name python=3.6
 
//**环境
source activate env_name
 
//退出环境
source deactivate
 
//删除环境
conda remove -n env_name --all
 
//查看conda环境
conda info --env
 
//查看conda的安装包
conda list

二、cuda与cudnn的安装(使用conda)

1.首先,查看自己的驱动型号了,cuda9.0必须要Nivdia 384及以上才可以正常安装。
具体的过程,可以参考知乎,

Ubuntu16.04下cuda9.0+cudnn7.0安装

2.其次,进入创建的虚拟环境下

source activate env_name

使用conda安装cuda和cudnn可以保证不影响系统默认环境的cuda和cudnn

因为conda的默认源速度慢且没有cuda版本,因此需要添加国内清华或者中科大的源

//终端加入
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
 
//或者直接打开文件加入
sudo gedit ~/.condarc
//安装cuda
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
 
//安装cudnn
conda install cudnn=7.1.2

参考:https://blog.csdn.net/weixin_40609307/article/details/89205511