我想创建一个docker镜像来运行我需要在GPU上运行DNN的代码。在Dockerfile中,我想使用pip安装requirements.txt。服务器上安装了CUDA 10.1,我发现https://pytorch.org/get-started/previous-versions/的torch版本与CUDA 10.1兼容。我尝试了所有的版本一个接一个,但我遇到了一个错误的任何时间。例如,一次尝试的需求文件包括:
pandas
matplotlib
swig
torch==1.4.0 -f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
错误是:
ERROR: Could not find a version that satisfies the requirement torch==1.4.0 (from versions: 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1)
ERROR: No matching distribution found for torch==1.4.0
有帮助吗?
1条答案
按热度按时间4szc88ey1#
PyTorch 1.4.0支持的最高Python版本是3.8,如https://download.pytorch.org/whl/torch_stable.html所示。这就是为什么你不能在Python 3.9中安装它。尝试安装PyTorch 1.7.1,如错误消息所建议。