Linux pytorch安装 cuda,torch,torchvision版本选择


安装好anaconda后,安装pytorch:

(1)使用pip install** ,代替conda install***更方便

(2)需要用到GPU资源的话,要加cuda的内容。linux查看本机cudu版本的命令为:cat /usr/local/cuda/version.txt

(3)知道了cuda的版本,去查看对应torch和torchvision的版本

 

 我的cuda版本是10.0.130,所以我要安装的pytorch版本是1.1,1.2这些。我选择1.2版本,那torch对应也是1.2版本,torchvision就选择0.4.0版本。

(4)直接在shell里输入:pip install torch==1.2.0 torchvison==0.4.0,回车,等他运行结束

(5)shell键入“python”进入python环境,然后输入“import torch",看看是否成功,”print(torch.cuda.is_available()) ”查看cuda版本是否对应。如果没有按照上面的版本对应的话,他会返回“False”,对应上了,就返回“True"。

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM