您的位置:首頁/技術文章
文章詳情頁

python 如何查看pytorch版本

瀏覽:30日期:2022-06-24 16:01:31

看代碼吧~

import torchprint(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 pandasconda install scikit-learnconda install torchnetexp:安裝pyorch1.0 + cuda9.0conda 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.whlpip3 install torchvision三.檢查&查看版本

1.檢查

import torchimport torchvisionprint(torch.cuda.is_available())

2.查看版本

import torchprint(torch.__version__)四.可能會遇到的問題:

安裝pip3

步驟:

第1步:安裝setuptools

wget --no-check-certificate https://pypi.python.org/packages/source/s/setuptools/setuptools-19.6.tar.gz#md5=c607dd118eae682c44ed146367a17e26tar -zxvf setuptools-19.6.tar.gzcd setuptools-19.6python3 setup.py buildpython3 setup.py install

第2步:安裝pip3

wget --no-check-certificate https://pypi.python.org/packages/source/p/pip/pip-8.0.2.tar.gz#md5=3a73c4188f8dbad6a1e6f6d44d117eebtar -zxvf pip-8.0.2.tar.gzcd pip-8.0.2python3 setup.py buildpython3 setup.py install

以上為個人經驗,希望能給大家一個參考,也希望大家多多支持好吧啦網。如有錯誤或未考慮完全的地方,望不吝賜教。

標簽: Python 編程
相關文章:
国产综合久久一区二区三区