RTX A4000 cuda 10.2 pytorch 应该用多少
时间: 2024-02-02 20:02:51 浏览: 150
MX250显卡驱动+cuda10.2+cudnn10.2
如果您的 RTX A4000 显卡使用的是 CUDA 10.2,那么您可以安装 PyTorch 1.7.1 或者更早的版本,这些版本支持 CUDA 10.2。您可以使用以下命令安装 PyTorch:
```
pip install torch==1.7.1+cu102 torchvision==0.8.2+cu102 torchaudio===0.7.2 -f https://download.pytorch.org/whl/cu102/torch_stable.html
```
请注意,如果您使用的是 Anaconda 或者其他类似的环境管理工具,也可以通过相应的命令安装 PyTorch 版本。最好在 PyTorch 官方网站上查看它们的版本兼容性文档以获取更多信息。
阅读全文