Linux下cuda10.0安装Pytorch和Torchvision|简记

---

服务器配置：RTX 2080 8G cuda10.0

---

---

这里安装的 版本为 pytorch1.1.0 torchvision0.3.0 ，使用conda 可以进行正确安装，
安装命令来自 pytorch 官方.
正确可用的安装方法如下：（亲测可用）

---

conda create -n nice python=3.6.9

conda activate nice

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

pip install opencv-python

pip install scipy

pip install thop

---

备注： 针对 cuda10.0 ,如果使用 pip 来安装 torch， 我遇到的报错如下：

---

NiceGAN代码运行报错如下：

RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:117

我认为的原因是，缺少了 cuda10.0 对应的 cudatoolkit 内核；
针对cuda10.0而言 ，pytorch 安装，使用 pip install torch==1.1.0 、pip install torchvision==0.3.0来直接安装导致该错误；

---

---

这里安装的 版本为 pytorch1.4.0 torchvision0.5.0 ，使用conda 可以进行安装，
安装命令来自 pytorch 官方.

---

conda create -n torch14 python=3.6.6

conda activate torch14

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.0 -c pytorch

pip install opencv-python

pip install scipy

pip install thop

py3.6_cuda10.0.130_cudnn7.6.3_0

---