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