Linux服务器下用conda配置pytorch+cuda

Linux服务器下配置pytorch+cuda

• 查看版本

首先查看服务器cuda支持的版本，

使用 **nvcc -V** 命令查看版本信息

• 创建虚拟环境
  Anaconda创建环境：
  比如，创建pyhon＝3.6的版本环境取名叫 pytorch_gpu

conda create -n  pytorch_gpu python=3.6

下载完毕后；
激活虚拟环境

source activate pytorch_gpu   （pytorch_gpu是你自己命名的虚拟环境名字）

• 为服务器配置镜像
  如果你已经配置了清华的镜像，不用理睬，或者可以接受下载速度慢的问题，也可以跳过这一步。

清华源在2018年4月1号推出了一键tuna的服务，我们只需要下载相关的脚本即可。进入清华源tuna服务的官网：https://tuna.moe/oh-my-tuna/，可以看到几行命令，使用wget命令将oh-my-tuna.py脚本下载到本地；
进入官网你会看到如下界面：

将命令复制。

wget https://tuna.moe/oh-my-tuna/oh-my-tuna.py

sudo python oh-my-tuna.py --global

• 进入pytorch官网
  网址：https://pytorch.org/
  
  然后找到你的服务器支持的版本（千万别选错了），复制对应的命令：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

接下来就是静等下载了，