1.安装或更新显卡驱动
根据自己的电脑的配置和需求,安装或更新显卡驱动,nvidia显卡驱动下载地址:官方驱动 | NVIDIA
下载可能比较慢,有梯子的话会好些。下载完成后安装exe文件即可。安装完成后,打开cmd命令窗口,输入如下的命令,可查看显卡驱动信息。
nvidia-smi
根据显卡驱动信息,可以看到驱动的版本和最高支持的CUDA版本。然后,我们就需要根据这个信息来安装环境。
2.安装Anaconda
我们将通过Anaconda安装pytorch。Anaconda官网地址: