时间:2022-04-21 09:45:29 | 栏目:Python代码 | 点击:次
看代码吧~
import torch print(torch.__version__)
补充:pytorch不同版本安装以及版本查看
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
安装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