我试图在我的Windows 11机器上安装支持CUDA的PyTorch，它安装了CUDA 12和Python 3.10。当我运行nvcc --version时，我得到以下输出：
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Aug_15_22:09:35_Pacific_Daylight_Time_2023
Cuda compilation tools, release 12.2, V12.2.140
Build cuda_12.2.r12.2/compiler.33191640_0我想安装支持CUDA的PyTorch 2.0.0版，所以我尝试运行以下命令：
python -m pip install torch==2.0.0+cu117但是，我遇到了以下错误：
ERROR: Could not find a version that satisfies the requirement torch==2.0.0+cu117 (from versions: 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==2.0.0+cu117有人有什么建议吗？
1条答案
按热度按时间jum4pzuy1#
要使用pip或conda安装PyTorch，并不强制要求在系统中本地安装nvcc（CUDA运行时工具包）;你只需要一个兼容CUDA的设备。
要安装PyTorch（2.0.1 with CUDA 11.7），您可以运行：
对于CUDA 11.8，运行：
如果你想使用CUDA 12.1，你可以使用以下命令安装PyTorch Nightly：（仅供参考：截至今天（2023年9月9日），cu12.1无法用于稳定版本。
您可以在PyTorch的官方页面上找到更多详细信息：https://pytorch.org/get-started/locally/。
希望能帮到你谢谢你，谢谢！