pytorch安装

概述pip3installtorch===1.5.0torchvision===0.6.0-fhttps://download.pytorch.org/whlorch_stable.html-ihttps://pypi.tuna.tsinghua.edu.cn/simpleimporttorch报错报错ImportError:numpy.core.multiarrayfailedtoimportpipuninstallnumpypipinstall

 pip3 install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.HTML -i https://pypi.tuna.tsinghua.edu.cn/simple

import torch 报错

报错importError: numpy.core.multiarray Failed to import

pip uninstall numpypip install numpy

即可

测试cuda

import torchprint(torch.cuda.is_available())

NumPy 安装

sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose

NumPy更新

pip install --upgrade numpy 

@H_404_50@

python查看pytorch版本

import torchprint(torch.__version__)

换清华源下载pytorch

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --set show_channel_urls yesconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

pytorch官网——下载previous version
记住将最后的-c pytorch去掉！这样才是从清华源下载，否则是从官网下载，很慢很慢。

虚拟环境安装指定版本的pytorch安装

bash + 制定版本的Anconda.sh

创建虚拟环境，qq指的是你的虚拟环境名称（随意）

conda create -n qq python=3.6.1

进入虚拟环境qq

source activate qq

安装指定版本pytorch，我这里安装的是1.6版本

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1

注意，此时仅仅是在终端的torch版本改变成1.6而已！！！

还需要jupyter notebook中添加虚拟环境

conda install nb_conda

恭喜！这样你就可以在jupyter notebook中使用1.6版本的torch了

总结

以上是内存溢出为你收集整理的pytorch安装全部内容，希望文章能够帮你解决pytorch安装所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。