rtx3090对应的pytorch版本
时间: 2023-09-23 16:08:33 浏览: 172
【环境配置】RTX 3090+cuda11.2+python39+pytorch1.8.0(csdn)————程序.pdf
5星 · 资源好评率100%
RTX 3090 可以使用 PyTorch 的最新版本进行训练和推理。PyTorch 版本的选择主要取决于你的操作系统和 CUDA 版本。目前,PyTorch 1.8.0 是最新的稳定版本。你可以通过以下命令来安装 PyTorch:
```
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
请注意,这是针对 CUDA 11.1 和 cuDNN 8.0 的版本。如果你的 CUDA 版本不同,可以适当修改以上命令中的版本号。
同时,请确保你已经安装了与 PyTorch 版本兼容的 NVIDIA 驱动程序和 CUDA 工具包,以便正确地利用 RTX 3090 的性能。
阅读全文