m0_61724130的博客

原创 适配于CUDA的PyTorch安装

需要安装与你的GPU对应的CUDA工具包，并且使用编译过CUDA支持的PyTorch版本。使用他给的命令就可以安装与CUDA相适应的PyTorch，注意这里要更改你需要的CUDA版本号。2、查看与你的GPU适配的CUDA，在终端或命令窗输入nvcc --version。本文章是本人在进行基于gan的神经网络模型训练过程中遇到的一些问题的总结。就能看到对应的CUDA版本，这里是11.7，那么就在官网下载对应版本就好了。之后你的终端就会展示出带有torch的软件包。二、查看你的电脑适配的CUDA。