复现一个代码,需要的torch==1.8.1+cuda90.cudnn7.6.5的要求应该怎样下载
时间: 2024-09-14 08:13:53 浏览: 35
要复现一个需要特定版本的PyTorch和CUDA、cuDNN环境的代码,你可以按照以下步骤进行操作:
1. 安装特定版本的CUDA和cuDNN:
- CUDA 9.0下载地址:https://developer.nvidia.com/cuda-90-download-archive
- cuDNN 7.6.5下载地址:https://developer.nvidia.com/rdp/cudnn-archive
下载与你的操作系统对应的版本,并根据官方指南安装。
2. 安装特定版本的PyTorch:
- 首先,访问PyTorch官网的安装指南页面:https://pytorch.org/get-started/locally/
- 在该页面上,选择你的系统环境、包管理器(例如conda或pip)、CUDA版本(对于你的情况,选择CUDA 9.0)。
- 页面将提供给你一个安装命令,例如使用conda时可能看起来像这样:
```
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=9.0 -c pytorch
```
- 执行该命令,conda将为你安装所需版本的PyTorch及其依赖项。
3. 验证安装:
- 安装完成后,你可以通过运行以下Python代码来验证PyTorch和CUDA的安装是否正确:
```python
import torch
print(torch.__version__)
print("CUDA available: ", torch.cuda.is_available())
```
确保在安装过程中检查你的系统环境是否支持所选的CUDA版本,并且你的NVIDIA驱动版本满足CUDA 9.0的安装要求。
阅读全文