如何安装指定版本Pytorch:
#使用conda安装指定版本
conda install pytorch=0.1.10 -c soumith
#使用pip安装指定版本
pip install pytorch==0.1.10
如何查看当前pytorch版本:
import torch
print(torch.__version__)
验证CUDA安装是否成功:
#找到 /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery 文件夹下输入:
sudo make
./deviceQuery