1,安装anaconda
2,查看显卡版本信息
3,安装对应的驱动程序cuda tooltik
4,安装pytorch
- 首先安装anaconda ,安装anaconda的目的在于创建虚拟环境,因为有的包需要不同的python版本,但是你不能在一个环境内配置多个python,所以就需要虚拟环境的存在。
- 查看显卡的信息:
进入显卡版本的目录:
cd C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi
最上面一行显示的版本号与cuda版本。
-
cuda toolt