CuDa,CuDNN,GPU版本的pytorch及使用教程

Step1.CuDa和CuDNN的安装教程

注意点：cuda版本和cudnn版本要相匹配，且不要安装太高版本的，不稳定

见：Cuda和cuDNN安装教程(超级详细)-优快云博客

Step2.验证CuDa和CuDNN是否安装成功

验证很重要，要不然后面白费，还得重来

见：Windows10检查Cuda和cuDNN是否安装成功？_查看大cuda时,cmd终端输入nvcc -v命令显示不是内部命令或外部命令,这是为什么-优快云博客

Step3.安装gpu版本的pytorch

1.要查看电脑是否有显卡

2.建议python环境

conda create -n py39 python=3.9

记得查看自己的python版本号：

python --version

3.选择合适cuda版本的pytorch

pytorch网址：

复制下方命令：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

4.验证是否安装成功：

显示true即为安装成功！

5.注意

卸载pytorch代码：

pip uninstall pytorch

在pycharm中使用gpu版本的pytorch：新建project，选择torch路径，注意选择python.exe