1.1下载cudnn,解压
1.2.找到本级cuda安装路径
1.3.刚才解压文件复制到cuda安装目录
2.1 安装pytouch
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
3.pytouch验证
我这儿是有问题的 PS C:\Users\Administrator\PycharmProjects\pythonProject>
验证是否安装成功
在命令行左边为 pytorch 环境中,输入 python
之后,输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了
接下来,输入 torch.cuda.is_available(),如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU
参考:https://blog.csdn.net/weixin_43507693/article/details/109015177
PS 开头
4.pycharm中的terminal运行前面的PS如何修改成pytorch环境
5.idea初始化工具 提示 Please select a valid Python interpreter
interpreter