cuda 10.0 安装pytorch
时间: 2023-09-30 14:03:23 浏览: 185
windows10安装GPU版pytorch详细教程(以cuda10.0+pytorch1.1为例)
根据提供的引用内容,您可以使用以下步骤来安装CUDA 10.0对应的PyTorch版本:
1. 首先,打开终端并输入以下命令来安装PyTorch和Torchvision:
```
pip install torch==1.7.1 torchvision==0.8.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/
```
2. 等待安装完成后,在终端输入`python3`,然后输入`import torch`来验证是否安装成功。
请注意,以上步骤仅适用于CUDA 10.0版本。如果您使用的是其他版本的CUDA,请根据对应版本的PyTorch官方文档进行安装。
阅读全文