- 查看电脑显卡配置,若同时存在集成显卡和NVIDIA独立显卡,在NVIDIA界面将自动分配换为高性能,同时查询显卡所支持的cuda版本
- 安装anaconda,cuda,cudnn:注意需三者需统一python和cude版本
- 在anaconda中创建虚拟环境,在离线模式下下载pytorch:https://download.pytorch.org/whl/torch_stable.html(注意cuda及python版本一致)
(可参考该篇博文:Pytorch安装,这一篇就够了,绝不踩坑_孔德浩的博客-CSDN博客_pytorch 安装)
- 下载好两个whl文件后,用anaconda Prompt切换到你的虚拟环境,在虚拟环境中切换到你两个whl文件的安装目录,在文件名前加pip install去安装这两个文件。
- 检验:
conda activate pytorch
python
import torch
print(torch.cuda.is_available())
pycharm中pytorch环境配置
打开pycharm项目,file-settings-Project:-project Interpreter-Python 3.9(pytorch),选择在anaconda中创建的虚拟环境pytorch中python.exe,即可使用pytorch