请参考我这篇文章：Ubuntu16.04安装cuda及cudnn方法
torch官网上可找到各种安装方式，这里选择whl安装，首先在网站上下载对应cuda10.0版本的whl文件。这里我下载的是1.0.1.post2(一个项目指定要用这个版本，:<我也没办法)，又由于我用的python3.5，因此选择下载torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl这个安装。
我要评论sudo chmod a+x torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip install torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
另外还需安装torchvision，这里对应的版本是0.2.2.post2，用pip安装：
pip install torchvision==0.2.2.post2 -i https://pypi.tuna.tsinghua.edu.cn/simple/
测试是否安装成功：
import torch
print(torch.__version__) #输出Pytorch的版本
print(torch.cuda.is_available()) #看看能否正常调用CUDA
一切正常，大功告成。
本文地址：https://blog.csdn.net/wolfcsharp/article/details/107474382
如对本文有疑问， 点击进行留言回复！！
【状压dp】[HDU1400 & poj2411] Mondriaan‘s Dream
网友评论