pytorch1.4.0安装哪个cudann
时间: 2023-09-29 14:10:54 浏览: 81
如果你的GPU型号是 NVIDIA 的 GeForce GTX 10 系列或 Tesla V100,可以安装 cuDNN 7.6.5 for CUDA 10.1。
如果你的GPU型号是 NVIDIA 的 Tesla P100,可以安装 cuDNN 7.6.5 for CUDA 10.0。
如果你的GPU型号是 NVIDIA 的 GeForce GTX 9 系列,可以安装 cuDNN 7.0 for CUDA 9.0。
具体情况可以参考 NVIDIA 的官方文档:https://docs.nvidia.com/deeplearning/sdk/cudnn-support-matrix/index.html
相关问题
pytorch1.4.0安装
以下是安装PyTorch 1.4.0的步骤:
1. 首先,确保你已经安装了Python和pip。你可以在命令行中运行以下命令来检查它们的版本:
```shell
python --version
pip --version
```
2. 接下来,使用pip安装PyTorch 1.4.0。在命令行中运行以下命令:
```shell
pip install torch==1.4.0
```
3. 如果你的网络环境无法直接从PyTorch官方源下载安装文件,你可以尝试使用离线安装文件进行安装。根据引用提供的离线安装文件进行操作。
4. 安装完成后,你可以在Python中导入PyTorch并验证版本。在Python交互环境中运行以下代码:
```python
import torch
print(torch.__version__)
```
如果输出为1.4.0,则表示PyTorch 1.4.0已成功安装。
pip 安装pytorch1.4.0 gpu
您可以使用以下命令在GPU上安装PyTorch 1.4.:
```
pip install torch==1.4.+cu100 torchvision==.5.+cu100 -f https://download.pytorch.org/whl/cu100/torch_stable.html
```
其中,`cu100`表示CUDA 10.版本,如果您的CUDA版本不同,请相应更改。如果您没有安装CUDA,则需要安装CPU版本的PyTorch。
阅读全文