Ubuntu 16.04上源码编译和安装pytorch教程,并编写C++ Demo CMakeLists.txt | tutorial to compile and use pytorch on...
本文首发于个人博客 https://kezunlin.me/post/54e7a3d8/ ,欢迎阅读最新内容! tutorial to compile and use pytorch on ubuntu 16.04 PyTorch for Python install pytorch from anaconda conda info --envs conda activate py35 # newest version # 1.1.0 pytorch/0.3.0 torchvision conda install pytorch torchvision cudatoolkit=9.0 -c pytorch # old version [NOT] # 0.4.1 pytorch/0.2.1 torchvision conda install pytorch=0.4.1 cuda90 -c pytorch output The following NEW packages will be INSTALLED: pytorch pytorch/linux-64::pytorch-1.1.0-py3.5_cuda9.0.176_cudnn7.5.1_0 torchvision pytorch/linux-64::torchvision-0.3.0-py35_cu9.0.176_1