杂记 ubuntu16.04+pytorch0.4.1

1 查看本机python版本号

python -V

 得到

Python 3.6.5 :: Anaconda, Inc.

2 使用anaconda建立新的虚拟环境，在其中安装pytorch

conda create -n 环境名 python=3.6

3 进入该虚拟环境

source activate 环境名

指令前面会多一个（环境名）

4 安装pytorch

conda install pytorch=0.4.1 torchvision=0.2.1 -c pytorch

至此pytorch安装完成

查看pytorch版本

先输入

python

进入python运行状态

再输入

import torch
print(torch.__version__)

输出pytorch版本号

 

若要离开该虚拟环境

source deactivate

安装tensorboard

pip install tensorboardX==1.4

查看cuda版本

cat /usr/local/cuda/version.txt
或
nvcc -v

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

查看opencv库

pkg-config opencv --libs

查看opencv版本号

pkg-config opencv --modversion