以下所有安装均是默认路径
- 安装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==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
终端输入:
python
再输入:
import torch
如果没有输出,代表安装成功;
- 在虚拟环境下,终端输入以下命令可以安装想要的插件:
pip install pandas
参考博客:
神经网络(十五)在VS Code下搭建PyTorch环境
Anaconda超详细安装教程(Windows环境下)
windows下sklearn库的安装教程,Numpy、Scipy、matplotlib