以下所有安装均是默认路径
- 安装cuda(确定自己的显卡所支持的CUDA版本,我的电脑cuda版本是11.7):
cuda下载地址 - 安装vscode(最新版本):
vscode下载地址 - 安装anaconda(最新版本):
anaconda下载地址 - 电脑搜索栏搜索: Anconda Prompt,点击进入终端;
- 输入:
conda create -n test python=3.11.4
,就可以创建名为test,python版本为3.11.4的虚拟环境(我下载的conda对应的python版本为3.11.4); - 输入:
conda activate test
,就可以进入名为test的虚拟环境; - 对应你电脑的cuda版本,安装对应的pytorch:
pytorch安装命令
在虚拟环境下,终端输入以下命令可以安装pytorch:
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio