torchvision 和 pytorch 安装

conda install pytorch torchvision cuda92 -c pytorch
conda install pytorch cuda92 -c pytorch

特别注意:

版本一定要与电脑中的cuda版本对应

完成之后可在python中查看

import torch
import torchvision
print(torch.__version__)#输出PyTorch版本

 

你可能感兴趣的:(Python)