安装cuda113对应的torch和torchvision
时间: 2023-11-20 15:40:05 浏览: 156
安装CUDA 11.3:
1. 访问 NVIDIA 官网下载 CUDA Toolkit 11.3,选择对应操作系统和计算能力的版本,下载后进行安装。
2. 添加环境变量,在系统环境变量中添加 CUDA_HOME 和 PATH 两个变量,分别指向 CUDA 安装目录和 CUDA 的 bin 目录。
安装 PyTorch 和 torchvision:
1. 打开终端或命令行窗口,输入以下命令安装 PyTorch:
```
pip install torch==1.9.0+cu113 torchvision==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
```
2. 安装完成后,可以通过以下代码检查 PyTorch 是否成功安装:
```
import torch
print(torch.__version__)
```
如果输出版本号,则说明 PyTorch 安装成功。
3. 安装 torchvision,只需要输入以下命令:
```
pip install torchvision
```
4. 安装完成后,可以通过以下代码检查 torchvision 是否成功安装:
```
import torchvision
print(torchvision.__version__)
```
如果输出版本号,则说明 torchvision 安装成功。
阅读全文