Linux下安装Anaconda,pytorch-0.4.1-GPU,tensorboard

Linux下安装Anaconda,pytorch-0.4.1-GPU,tensorboard

  1. 首先:查看cuda的版本:cat /usr/local/cuda/version.txt
    在这里插入图片描述
  2. 去官网下载对应版本的Anaconda,这里我要用的是python 3.6.4,因此我下载的安装包为:anaconda3-5.0.1-Linux-x86_64.sh
    在这里插入图片描述
    python版本对应的Anaconda如下:
    Linux下安装Anaconda,pytorch-0.4.1-GPU,tensorboard_第1张图片
  3. 安装Anaconda的话:bash Anaconda3-5.2.0-Linux-x86_64.sh,然后一路enter和yes就好了。最后的VSCode 那一步,我装了好久也没成功,报了没连接网络的错,但是网是正常的,后来我直接no了,发现不影响pytorch的使用。
  4. 配置Anaconda的环境,一般来说会自动帮你配好,可以查看一下,你的账号下的.bashrc文件的末尾有没有Anaconda的路径。如下:没有则手动添加。
    在这里插入图片描述
  5. 安装好之后,创建一个虚拟环境来配pytorch,方便你日后要装tensorflow等,这样创建的虚拟环境不会发生冲突。
conda create -n pytorch python=3.6

当然文件夹的名字可以自行定义了。然后输入以下进行激活环境:

source activate pytorch

在这里插入图片描述
想要退出环境,则:source deactivate

在这里插入图片描述

  1. 接下来就是安装pytorch的GPU版本,这里要注意,我的cuda为9.0,因此在pytorch的官网,我去选择适合自己的版本,然后输入对应的命令进行安装。
    Linux下安装Anaconda,pytorch-0.4.1-GPU,tensorboard_第2张图片
    在终端输入:conda install pytorch=0.4.1 cuda90 -c pytorch
    然后一路y就好了,安装的下载过程可能有点慢,中断就重新来几次就好了。
  2. 安装好pytorch之后,一般需要配合tensorboard进行可视化。因此可以直接通过conda install tensorflow 进行安装,通过安装tensorflow会直接安装对应的tensorboard。安装完成后就可以直接调用了。

你可能感兴趣的:(计算机视觉)