安装pytorch=1.4.0 torchvision=0.5.0 cudatoolkit=10.0 -c pytorch

背景

最近在复现一篇论文----stargan-v2

按照要求，clone下来代码，并创造stargan-v2的环境，py=3.6.7.

然后安装第一个数据包pytorch以及torchvison，cudatoolkit(在这之前以及安装好了cuda=10.0)

安装第一个数据包

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

然而出现以下报错

报错原因是安装包之间存在冲突。

解决方案；

使用指令conda-forge，解决了冲突问题（这里先安装pytorch与cudatoolkit）

conda install -c conda-forge pytorch=1.4.0  cudatoolkit=10.0 

安装成功

然后继续用conda-forge，安装torchvision=0.5.0，

conda install -c conda-forge torchvison=0.5.0

却也报错;

将conda 改为 pip 继续运行上面代码

pip install  torchvision==0.5.0

安装成功。

检查

检查安装列表，发现数据包在里面
激活python 调用pytorch发现没有报错
安装成功！

如果你尝试用我的方法，却没有成功，或许可以先更新conda版本；
conda update -n base conda

最后，祝你好运。
（如果你也对stargan感兴趣，可以一起交流交流）