对于ubuntu16.04下，通过conda安装cuda和cudnn。以及安装GPU版的tensorflow和pytorch，可以参考以下博客。
ubuntu16.04 通过anaconda建立虚拟环境，安装cuda9.0，cudnn7.1.2

虚拟环境下，基于cuda9.0，cudnn7.1.2安装tensorflow和pytorch

一.anaconda的安装

1.首先查看想要安装的python版本对应的anaconda版本
Anaconda版本与Python版本的对应关系

2.下载anaconda

• （1）Anaconda 安装包可以到

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

下载。

• （2）也可以去官网下载

https://www.anaconda.com/download//

• （3）下载完成后，进入下载的文件夹，打开终端

然后运行以下命令，例如：我下载的是Anaconda3-5.0.0-Linux-x86_64.sh，所以命令如下：

bash ./Anaconda3-5.0.0-Linux-x86_64.sh

全程 yes +回车 ，默认安装在home目录下

• （4）检查环境变量，没有则添加

查看命令

sudo gedit ~/.bashrc

• （5）检查末尾是否已经添加如下变量，没有则添加

export PATH="/home/你的用户名/anaconda3/bin:$PATH"

二.pycharm的安装

具体安装可以参考下面这篇博客：
Ubuntu 16.04 安装 PyCharm

anaconda安装参考博客：
https://blog.csdn.net/luojie140/article/details/78696330