Pytorch安装
安装是真的麻烦,太麻烦了,菜鸡颤栗。
我是在anaconda下面安装的,首先吧还是先创建一个虚拟环境,这里配置的环境是python3.6.9下,可以根据自己需求进行重新选择配置:
conda create -n torch_env python==3.6.9
为了避免各种各样的墙导致下载速度过慢,我们先把conda和pip换源:
conda 换源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip 换源:
这里pip最好版本在10.0.0以上吧。
pip install pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu,cn/simple
创建好虚拟环境之后就是安装乱七八糟的一堆包了,这里用conda或者pip都完全没得问题:
conda install spyder
conda install cudatoolkit==8.0
conda install cudnn==6.0
pip install scipy
pip install matplotlib
pip install opencv-python
pip install scikit-image
pip install keras==2.1.3
pip install https://download.pytorch.org/whl/cu80/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip install torchvision
到这里之后就是疯狂试探了,讲道理我也没搞清楚什么问题,我就用这个方法搞定了,大家可以试试,欢迎一起讨论啊。