已经通过 conda install pytorch 安装过cpu版本
安装GPU版本
安装CUDA
https://developer.nvidia.com/cuda-toolkit 下载最新版本的CUDA
需要先安装visual studio(安装CUDA的过程中会提示)
https://visualstudio.microsoft.com/zh-hans/ 下载visual studio并安装
然后继续安装CUDA即可,安装结束后命令行输入 nvcc -V 成功显示CUDA信息
安装gpu版本pytorch
安装前
import torch
torch.cuda.is_available()
output: False
https://pytorch.org/get-started/locally/
conda找不到torchvision ,尝试用pip安装
pip3 install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
命令行的这条命令最后显示成功安装了torch和torchvision,torch.cuda.is_available()为True
但是在conda的虚拟环境中并没有被成功安装。
解决方法是保证这个虚拟环境下应该安装了pip,然后运行
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
成功后
import torch
torch.cuda.is_available()
output: True
参考: