anaconda与pycharm应用pytorch步骤

安装pytorch

官网下载CUDA
CUDA 10.1版本
命令行下 看 cuda是否安装成功

下载后的CUDNN

检验CUDNN安装是否成功

此处进入窗口

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite

.\bandwidthTest.exe
.\deviceQuery.exe



CUDA CUDNN完成

配置anaconda pytorch 虚拟环境打开Anaconda Prompt

查看安装情况

envs_dirs:
  - D:\360Downloads\Anaconda3\envs
pkgs_dirs:
  - D:\360Downloads\Anaconda3\pkgs

CUDNN 对应

我上面查找的我的CUDA版本为11.4，那么我可以安装CUDA11.4以下的版本的pytorch包，这里我就选择11.6。

安装pytorch到anaconda

anaconda 配置 新的虚拟环境

conda create -n pytorch（pytorch名字） python=3.10   // ame名字   我的
conda create -n pytorch python=3.10 //这里我使用的是3.8版本的python  对

激活环境

activate ame

1. 进入相应环境

conda activate  ame

2. 安装pytorch

# CUDA 11.3
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

//首先进入pytorch环境，若已进入忽略
conda activate pytorch
python
>>>import torch
//输入完成后如果没有报错，则证明你已经成功安装了恭喜你！！！
>>>torch.cuda.is_available()
true
//继续输入上面代码，输出True，则意味着你可以使用 GPU

查看下载渠道和环境安装路径等信息
conda info
pytorch 与pycharm连接

打开相应文件 在jupyter notebook中使用pytorch

activate ame 
cd ...

问题：
使用anaconda3安装了pytorch，在anaconda prompt中激活新创建的环境之后，import torch是没有问题的，如下图：

解决：

打开Anaconda Prompt，激活安装pytorch的ame环境变量，在此环境变量中安装jupyter notebook，命令如下：

conda activate ame
conda install ipython
conda install jupyter

再次进入