找不到 CUDA 10.0 的 Pytorch 1.7 兼容版本

Question

我正在重现一个需要 pytorch 1.7 和 CUDA 10.0 的项目。 但是，我发现（https://pytorch.org/get-started/previous-versions/）CUDA 10.0 没有 pytorch 1.7 版本。

这种情况，安装cudatoolkit 10.0后该怎么办？

Answer 1

您绝对不会在 'https://pytorch.org/get-started/previous-versions/' 找到 Pytorch 1.7 版本（可能会检查 git，如果可用的话克隆它）在 Conda 中创建新环境的最佳方法，将您的所有要求纳入requirements.txt并使用pip install -rrequirements.txt安装它们或使用conda install pytorch=1.7 torchvision=0.8 cudatoolkit=10.0 -c pytorch。希望它对您有用。