CUDA, CUDNN & Installation

Cyan

2023-06-24

注意

本文最后更新于 2023-06-24，文中内容可能已过时。

Create Python Environment

1

cond create -n env_name python=3.x

1
2


import torch
torch.cuda.is_available()

1

conda install cudatoolkit=x.x

简单区分：CUDA为NV的计算框架，而CUDNN为NV为深度学习设计的计算库

建议使用pytorch官方的conda指令一键安装