用anaconda测试cuda与cudnn
时间: 2024-05-13 12:15:07 浏览: 69
使用Anaconda
如果你已经安装了Anaconda,可以按照以下步骤测试CUDA与cuDNN是否正确安装:
1. 打开Anaconda Prompt(Windows)或终端(Mac/Linux)。
2. 创建一个新的虚拟环境,并激活它:
```
conda create --name test_env
conda activate test_env
```
3. 安装必要的包,包括PyTorch、TensorFlow等:
```
conda install pytorch torchvision cudatoolkit=<version>
conda install tensorflow-gpu
```
注意:`<version>`需要替换为你安装的CUDA版本,例如`cudatoolkit=10.1`。
4. 运行以下Python代码:
```python
import tensorflow as tf
import torch
print("TensorFlow is using GPU: ", tf.test.is_gpu_available())
print("PyTorch is using GPU: ", torch.cuda.is_available())
```
如果输出结果都为True,则说明CUDA和cuDNN已经正确安装,并且可以在你的计算机上使用GPU进行加速。
希望这能帮助到你!
阅读全文