文章目录
- 1.安装nvidia cuda驱动
- CMD中看一下cuda版本:
- 下载并安装cuda驱动
- 2.创建虚拟环境并安装pytorch的torch_cuda
- 3.测试
- 附录
1.安装nvidia cuda驱动
CMD中看一下cuda版本:

注意: 红框的cuda版本,是你的显卡能装的最高的cuda版本,所以可以选择低于它的版本。比如我的是11.4,你可以选择11.3
下载并安装cuda驱动
https://developer.nvidia.com/cuda-toolkit-archive
点击