Linux服务器下配置pytorch+cuda
- 查看版本
使用 **nvcc -V** 命令查看版本信息
文章图片
- 创建虚拟环境
Anaconda创建环境：
比如，创建pyhon＝3.6的版本环境取名叫 pytorch_gpu
conda create -npytorch_gpu python=3.6
文章图片
文章图片
下载完毕后；
激活虚拟环境
source activate pytorch_gpu（pytorch_gpu是你自己命名的虚拟环境名字）- 为服务器配置镜像
如果你已经配置了清华的镜像，不用理睬，或者可以接受下载速度慢的问题，也可以跳过这一步。
进入官网你会看到如下界面：
文章图片
将命令复制。
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
文章图片
【Linux服务器下用conda配置pytorch+cuda】
文章图片
接下来就是静等下载了，
