目录
一、Miniconda的安装
首先进入conda的官网,下载miniconda,链接:Miniconda — miniconda documentation
执行安装程序,建议不要安装到C盘,我个人安装到了D盘。
二、Pytorch环境的配置
打开Anaconda Powershell Prompt,输入以下指令查看自己的CUDA version:
nvidia-smi
输入python 查看自己的python版本:
根据自己的python版本,创建虚拟环境:
conda create -n mypytorch python=3.11
进入虚拟环境:
conda activate mypytorch
打开pytorch官网,根据自己的cuda选择,注意只能选比自己的cuda低的,比如我的是12.0,我选的是11.8
在已经进入虚拟环境的条件下粘贴command并运行
enter运行开始下载,遇到选择输入y,最后会提示下载完成。
三、Pycharm中使用pytorch
我的Pycharm是2023年版本的
打开Pycharm,新建项目,添加本地解释器
点击conda环境,找到miniconda\Library\bin\conda.bat文件,点击加载环境,选择自己的虚拟环境,然后确定。
在创建好的文件中输入以下代码:
import torch
print(torch.cuda.is_available())
运行,可看到结果为true,完成 。