目录
桌面鼠标右键进入NVIDIA控制面板--帮助--系统信息--组件--查看cuda版本为11.3
在 Anaconda prompt 窗口中,复制官网提供的命令并输入(若想使用镜像源加速,需要去掉末尾的-c pytorch)
一.检查显卡与python版本
桌面鼠标右键进入NVIDIA控制面板--帮助--系统信息--组件--查看cuda版本为11.3
二.创建pytorch环境
进入 Anaconda prompt
命令窗口,输入
conda create -n pytorch python=3.8
-
进入pytorch官网:https://pytorch.org/
-
可看到下图,官网会自动根据你的电脑,显示你可安装的CUDA版本,并给出安装命令
- 激活pytorch环境,输入以下代码
activate pytorch
三.安装 PyTorch
在 Anaconda prompt
窗口中,复制官网提供的命令并输入(若想使用镜像源加速,需要去掉末尾的-c pytorch)
conda install pytorch torchvision torchaudio cudatoolkit=11.3
四.测试
输入python
,进入python
开发环境中
import torch
torch.cuda.is_available()
看到True, 终于成功