ubuntu18安装pytorch-gpu,一行conda命令安装,不需要单独安装cuda
生活随笔
收集整理的這篇文章主要介紹了
ubuntu18安装pytorch-gpu,一行conda命令安装,不需要单独安装cuda
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
1、準備條件:
- Anaconda3
- ubuntu18(其他版本應該也行,我電腦是ubuntu18)
- 已經安裝了顯卡驅動,如果沒有,就用下面的代碼安裝
安裝完顯卡驅動后記得重啟電腦
2、安裝pytorch-gpu
用conda命令
conda install pytorch-gpu torchvision這就會自動安裝pytorch-gpu和所需的cuda,cudnn。
測試 1
>>> print(torch.cuda.is_avilable())
Traceback (most recent call last):
? File "<stdin>", line 1, in <module>
AttributeError: module 'torch.cuda' has no attribute 'is_avilable'
>>> print(torch.cuda.is_available())
True
結果返回True,說明安裝gpu版成功
?
測試2,將torch放到gpu上。
>>> torch.rand(3,4).to('cuda')
tensor([[0.2400, 0.9294, 0.4117, 0.0969],
??????? [0.6468, 0.1129, 0.8767, 0.6132],
??????? [0.1432, 0.6973, 0.4343, 0.5509]], device='cuda:0')
結果:device='cuda:0',說明成功了
總結
以上是生活随笔為你收集整理的ubuntu18安装pytorch-gpu,一行conda命令安装,不需要单独安装cuda的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: (二)深度学习数据处理-----图片数据
- 下一篇: pytorch出现‘DataLoader