如何在Pipenv中指定pytorch / cuda版本

debugcn 发表于 Dev

萌

我正在尝试安装特定版本的pytorch，该版本与pipenv的特定cuda驱动程序版本兼容。该pytorch网站显示了如何将它与PIP：

pip3 install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html

我试图将其转换成这样的条目Pipfile：

[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/torch_stable.html"
verify_ssl = false

pytorch = {version="==1.3.1+cu92", index="pytorch"}
torchvision = {version="==0.4.2+cu92", index="pytorch"}

但是，这不起作用。无法解决此版本的依赖性。我不确定命令中与-f参数一起列出的网址pip3是否甚至是的有效来源pipenv。

我可以通过将命令传递给pip来安装两个库，如下所示：

pipenv run pip install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html

但是我对该解决方案并不真正满意，因为依赖项不在中，Pipfile因此我必须手动记录此命令的用法。

萌

上述方法的问题在于的结构https://download.pytorch.org/whl/torch_stable.html。Pipenv只能找到0.1到0.4.1的割炬版本，因为所有其他版本都将cuda（或cpu）版本作为前缀，例如cu92/torch-0.4.1-cp27-cp27m-linux_x86_64.whl。

但是cuda版本是一个子目录。因此，如果将源的URL更改为cuda版本，而仅在其依赖项中指定割炬版本。

[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu92"
verify_ssl = false

[packages]
torch = {index = "pytorch",version = "==1.4.0"}

我遇到的唯一问题是numpy无法识别为pytoch 1.4.0的依赖项。但这似乎是特定的火炬轮的问题。使用1.3.1或1.5.1版本以及最新的pipenv版本可以使用。

因此，如果使用进行安装后pipenv install，该命令pipenv run python -c "import torch"引发错误，则必须手动添加numpy。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。