请问CUDA11.4.153对应pytorch和torchvision的版本。以及安装Cuda和P有torch的pip代码
时间: 2023-09-28 20:07:36 浏览: 115
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
对于您的第一个问题,CUDA 11.4.153 对应的 PyTorch 版本是 1.9.0 和 torchvision 版本是 0.10.0。关于安装 CUDA 和 PyTorch,您可以参考以下代码:
安装 CUDA:
```bash
sudo apt-get update
sudo apt-get install cuda-11-4
```
安装 PyTorch 和 torchvision:
```bash
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
阅读全文