使用miniconda配置cuda10.2+pytorch1.10

在miniconda下配置环境遇到一些问题,为了以后方便复现,记录一下。

问题

开始从官网上下载了最新版本(4.10.3),然后使用

$ conda create -n cuda10.2 -f req.yml

安装环境时报错,问题应该是使用的miniconda版本和req.yml中定义的安装包的版本不兼容。
后改成

$ conda create -n cuda10.2

然后单独安装更高版本的包解决了这个问题。
但后面安装torch的时候还是有问题,pip和conda均无法成功安装。

解决方案

为了解决这个问题,降低了miniconda版本到3.9.2,从官网上之前的版本列表下载。
但仍然无法通过直接-f req.yml安装,还是通过下面流程逐一安装:

$ conda create -n cuda10.2
$ conda install cudatoolkit=10.2
$ conda install pip=21.0.1 // 默认为python=3.10.*, 这导致无法使用pip install torch安装torch 1.10
$ conda install python=3.8.5
$ pip3 install torch torchvision torchaudio // 将python降级到3.8.5之后使用该命令可以成功安装torch 1.10