解決pytorch-gpu 安裝失敗的記錄
ubuntu18 系統(tǒng)上已經(jīng)配置好gpu加速環(huán)境,
安裝pytorch-gpu 后出現(xiàn)以下問題:
import torch x = torch.Tensor(2,3) x_cuda = x.cuda()
x可以正常,當(dāng)定義x_cuda 后,命令行卡住,不再繼續(xù)輸出
解決方法正在尋找中。。。找到后將記錄解決方案。
幾周后終于找到了原因:
當(dāng)初pytorch-gpu為1.0版本,然而配置的cuda和cudnn版本較低,不支持高版本的pytorch,后來選擇安裝 0.4.1版本的pytorch,解決此問題。
當(dāng)前環(huán)境是 torch-0.4.1-cp36 +384顯卡驅(qū)動(dòng)+cuda8.0+cudnn6.0。
幾多艱辛,得以開顏。
補(bǔ)充:conda安裝pytorch-gpu版本下載出錯(cuò)的問題解決
換了清華的源之后,官網(wǎng)的命令是下面的:
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch
但是經(jīng)常會(huì)報(bào)HTTP error的錯(cuò)誤,將-c pytorch刪除即可
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python中交換兩個(gè)元素的實(shí)現(xiàn)方法
今天小編就為大家分享一篇Python中交換兩個(gè)元素的實(shí)現(xiàn)方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-06-06Pycharm無法使用已經(jīng)安裝Selenium的解決方法
今天小編就為大家分享一篇Pycharm無法使用已經(jīng)安裝Selenium的解決方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-10-10python網(wǎng)絡(luò)編程socket實(shí)現(xiàn)服務(wù)端、客戶端操作詳解
這篇文章主要介紹了python網(wǎng)絡(luò)編程socket實(shí)現(xiàn)服務(wù)端、客戶端操作,結(jié)合實(shí)例形式分析了python基于socket實(shí)現(xiàn)服務(wù)端、客戶端相關(guān)操作技巧與使用注意事項(xiàng),需要的朋友可以參考下2020-03-03Python+SeaTable實(shí)現(xiàn)計(jì)算兩個(gè)日期間的工作日天數(shù)
在實(shí)際的項(xiàng)目管理、任務(wù)管理、工作計(jì)劃等場(chǎng)景中,某些時(shí)間段會(huì)涉及雙休日、法定節(jié)假日,甚至還有公司自定義的工作時(shí)間安排,所以就需要計(jì)算出兩個(gè)日期間的實(shí)際工作日天數(shù)。本文用Python+SeaTable實(shí)現(xiàn)這一需求,需要的可以參考一下2022-07-07