首先安装anaconda,可以在以下镜像源网站进行下载
Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
接着对其环境变量进行配置,即搜索环境变量后打开path,将anaconda安装路径添加到变量中如下图所示
接着对pytorch进行安装,在开始中打开anaconda prompt,并输入conda list查看python的版本号,我的是3.12.4,然后输入命令conda create -n yenv python=3.12.4(yenv这个位置可以任意命名),接着弹出proceed,输入y即可安装成功。接着安装pytorch,接着进入刚创建的环境之中,即输入conda activate yenv;
此时进入pytorch官网,即pytorch.org进入后按照对应版本进行选择,复制下面的代码,即
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
返回anaconda prompt中在创建环境中直接粘贴,进行pytorch的安装,安装完毕后进入创建环境后先输入pyhon,然后输入以下代码进行测试
import torch
torch.cuda.is_available()
如图返回true即为安装成功。
最后在pycharm中进行pytorch的配置,打开pycharm后选择新建项目,选择自定义环境后选择conda,接着选择anacoda安装路径下scripts里的conda.exe,然后下拉环境选择刚创建的环境如图所示,接着创建好项目后就可以使用pycharm了。