1.查看自己的cuda
win+r 键:输入cmd
再输入:nvidia-smi
效果:CUDA Version:11.0
2.anaconda虚拟环境中安装 pytorch
激活虚拟环境后输入:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c conda-forge
3.查看是否安装成功
虚拟环境输入python
再输入import torch
退出exit()
win+r 键:输入cmd
再输入:nvidia-smi
效果:CUDA Version:11.0
激活虚拟环境后输入:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c conda-forge
虚拟环境输入python
再输入import torch
退出exit()