查看linux中pytorch的路径,python 如何查看pytorch版本

该篇博客详细介绍了如何使用conda和pip在不同的CUDA版本下安装PyTorch及其相关库,包括 torchvision 和其他依赖。提供了从创建环境到检查版本的完整步骤,并给出了可能遇到的问题及解决方法,适合初学者参考。
摘要由CSDN通过智能技术生成

看代码吧~

import torch

print(torch.__version__)

补充:pytorch不同版本安装以及版本查看

一:基于conda安装

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安装

安装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.检查

import torch

import torchvision

print(torch.cuda.is_available())

2.查看版本

import torch

print(torch.__version__)

四.可能会遇到的问题:

安装pip3

步骤:

第1步:安装setuptools

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

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

以上为个人经验,希望能给大家一个参考,也希望大家多多支持我们。如有错误或未考虑完全的地方,望不吝赐教。

时间: 2021-03-16

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值