1. 环境
anaconda+pycharm
(资源链接,及配置教程)
本文方法适用于单cuda版本配置,多版本的,仍在学习中。
2. windows步骤
- 升级invidia驱动
选择对应显卡 的驱动,直接下载安装即可。 - 安装CUDA,下载后直接安装。
- 安装pytorch
1)在页面下翻,选择对应选项
2)复制命令(run this command后的语句)
3)打开anaconda prompt对话框,先输入命令激活环境(想把pytorch安装在哪个环境里就激活哪个环境)命令:[ activate 环境名称]
4)把复制的命令粘过去等待安装完成,网络问题中断了就再执行一次。
3. linux
linux环境下也类似。