pytorch:不同版本的CUDA和CUDNN安装教程

通过Anaconda 安装

pytorch 是根据不同的cuda版本安装的

具体如下

cuda9.0

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

CUDA 8.x

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

CUDA 10.0

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

通过pip安装

python3

# Python 3.5
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

参考官网

你可能感兴趣的:(深度学习,强化学习,python小工具编写)