前提:安装好了适应版本的cuda
步骤:
1.打卡anaconda prompt——输入命令创建虚拟环境:
conda create -n pytorch python=3.8 anaconda
2. 激活环境:
conda activate pytorch
3. 安装pytorch:
第一种:官方命令直接下载
进入官网复制conda指令安装:
其他cuda版本指令:
Previous PyTorch Versions | PyTorch
第二种:先从官网下载资源,再pip安装
资源网址:https://download.pytorch.org/whl/torch_stable.html
下载完成后先激活环境再pip install 资源位置
conda activate pytorch
pip install D:\360Downloads\torch-1.10.0+cu111-cp38-cp38-win_amd64.whl
最后测试安装成功与否:
打开 Anaconda prompt
命令窗口,激活环境,输入python
,进入python
开发环境中
import torch
torch.cuda.is_available()
True