1. 更新显卡驱动(保证CUDA >= 10.2)(2021.8.31)
因为pytorch现在对CUDA有要求,有显卡的话需要保证CUDA大于等于10.2才行,没有显卡的可以跳过,到时候点CPU。
查看cuda可以在桌面右键打开NVIDIA控制面板,然后查看系统信息,点组件,如图:
如果cuda不够就需要更新显卡驱动,更新驱动用驱动软件或者官方驱动管理都可以。
2. 在Anaconda中创建新环境
首先你要先安装Anaconda,然后打开anaconda prompt:
conda create -n pytorch(环境名) python=