参考资料:
pytorch中文手册:https://github.com/zergtant/pytorch-handbook
因为本机已经安装过tensorflow了,故而在虚拟环境中安装,已经安装好anaconda3
#pytorch为环境名,这里创建python3.6版。
conda create -n pytorch python=3.6
# 切换到pytorch环境
activate pytorch
# ***以下为1.0版本安装***
#安装GPU版本,根据cuda版本选择cuda80,cuda92,如果cuda是9.0版,则不需要
#直接conda install pytorch -c pytorch 即可
# win下查看cuda版本命令nvcc -V
conda install pytorch cuda92 -c pytorch
# cpu版本使用
# conda install pytorch-cpu -c pytorch
# torchversion提供了conda的安装包,可以用conda直接安装了
# 安装方式如下:
# cpu版本
conda install pytorch-cpu torchvision-cpu -c pytorch
# GPU版
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
# cudatoolkit后跟着相应的cuda版本
网友评论