Ubuntu16.04安装pytorch方法
程序员文章站
2022-05-13 12:47:34
Ubuntu16.04 Anaconda3安装pytorch方法准备工作安装cuda10.0及cudnn7.4.下载pytorch安装文件安装pytorch准备工作安装cuda10.0及cudnn7.4.请参考我这篇文章:Ubuntu16.04安装cuda及cudnn方法下载pytorch安装文件torch官网上可找到各种安装方式,这里选择whl安装,首先在网站上下载对应cuda10.0版本的whl文件。这里我下载的是1.0.1.post2(一个项目指定要用这个版本,:<我也没办法),又由于...
Ubuntu16.04安装pytorch方法
准备工作
安装cuda10.0及cudnn7.4.
请参考我这篇文章:Ubuntu16.04安装cuda及cudnn方法
下载pytorch安装文件
torch官网上可找到各种安装方式,这里选择whl安装,首先在网站上下载对应cuda10.0版本的whl文件。这里我下载的是1.0.1.post2(一个项目指定要用这个版本,:<我也没办法),又由于我用的python3.5,因此选择下载torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl这个安装。
安装pytorch
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
上一篇: 微博官方提供的营销工具有哪些?好用吗
下一篇: 详解微博品牌营销攻略
推荐阅读
-
windows 7安装ORACLE 10g客户端的方法分享
-
mysql 5.7.17 winx64.zip安装配置方法图文教程
-
mysql 5.7.11 winx64.zip安装配置方法图文教程
-
WINDOWS 在安装WINDOWS ME过程中死机的解决方法
-
基于Fedora14下自带jdk1.6版本 安装jdk1.7不识别的解决方法
-
会声会影2018专业版怎么下载安装? 会声会影2018专业版的安装方法
-
PHPStudy下如何为Apache安装SSL证书的方法步骤
-
mysql 8.0.11 macos10.13安装配置方法图文教程
-
安装cad时电脑提示已经安装怎么办 cad无法安装的解决方法
-
解析windows下使用命令的方式安装mysql5.7的方法