环境配置
一、NVIDIA驱动安装与更新
首先查看电脑的显卡版本,步骤为:此电脑右击–>管理–>设备管理器–>显示适配器。就可以看到电脑显卡的版本,我的是3060显卡,
显卡驱动的下载链接
安装(更新)好了显卡驱动以后。我们按下win+R组合键,打开cmd命令窗口。输入如下的命令。
1 ;nvidia-smi
得到如下图的信息图,可以看到驱动的版本是537.42;最高支持的CUDA版本是12.2版本。得到显卡的最高支持的CUDA版本,我们就可以根据这个信息来安装环境了。
二、Anaconda 的安装
打开网址下载最新的版本就好
三:安装pycharm
打开pycharm,创建一个新的工程,
此时刚刚的右下角已经有了我们刚刚选择的base环境中的python了
四:打开pytorch的,由于开头我们通过驱动检测到我的显卡为 NVDIA GeForce 3060,最高支持cuda12.2版本,
将红色框下面的复制,然后打开新建的工程,再终端中输入这个,进行下载
(建议大家更换清华源后下载)
清华源:pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
然后会下载几个G的torch库,
下载完成后我们的环境就搭建成功了。