看代码吧~
|
1
2
|
import torch
print(torch.__version__)
|
补充：pytorch不同版本安装以及版本查看
一：基于conda安装
|
1
2
3
4
5
6
7
8
9
10
11
12
|
conda create --name pytorch_learn python=3.6.7#创建一个名为pytorch_learn的环境
source activate pytorch_learn #进入环境
conda install pytorch=0.3.1 cuda80 -c soumith #安装pytorch0.3.1+ cuda8.0（可自己指定）
conda install -c soumith torchvision #安装 torchvision
#安装常用的相关依赖库
conda install pandas
conda install scikit-learn
conda install torchnet
exp:
安装pyorch1.0 + cuda9.0
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
|
二.基于pip安装
|
1
2
3
|
安装cuda8.0+torch1.0+python3.6,可更改链接名指定版本
pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision
|
三.检查&查看版本
1.检查
|
1
2
3
|
import torch
import torchvision
print(torch.cuda.is_available())
|
2.查看版本
|
1
2
|
import torch
print(torch.__version__)
|
四.可能会遇到的问题：
安装pip3
步骤：
第1步：安装setuptools
|
1
2
3
4
5
|
wget --no-check-certificate https://pypi.python.org/packages/source/s/setuptools/setuptools-19.6.tar.gz#md5=c607dd118eae682c44ed146367a17e26
tar -zxvf setuptools-19.6.tar.gz
cd setuptools-19.6
python3 setup.py build
python3 setup.py install
|
第2步：安装pip3
|
1
2
3
4
5
|
wget --no-check-certificate https://pypi.python.org/packages/source/p/pip/pip-8.0.2.tar.gz#md5=3a73c4188f8dbad6a1e6f6d44d117eeb
tar -zxvf pip-8.0.2.tar.gz
cd pip-8.0.2
python3 setup.py build
python3 setup.py install
|
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/qq_33768643/article/details/93490713