命令行窗口中使用
conda create -n 環(huán)境名 python=所需python版本
即可創(chuàng)建虛擬環(huán)境
首先確定自己電腦的gpu版本
打開顯卡控制面板
點(diǎn)擊系統(tǒng)信息,選擇組件
產(chǎn)品名稱中CUDA后的11.0便是gpu版本
再在pytorch官網(wǎng)選擇對應(yīng)版本
進(jìn)入為pytorch創(chuàng)建的虛擬環(huán)境,輸入命令,等待片刻,pytorch便安裝完成了。
再在NVIDIA官網(wǎng)安裝cuda和cudnn
將cudnn的三個文件分別放入cuda安裝目錄下,即可完成配置
在命令行中輸入驗(yàn)證代碼
如上圖所示即為成功
到此這篇關(guān)于pytorch安裝及環(huán)境配置的完整過程的文章就介紹到這了,更多相關(guān)pytorch安裝及環(huán)境配置內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
標(biāo)簽:拉薩 亳州 黔東 廊坊 渭南 綿陽 興安盟 內(nèi)江
巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《pytorch安裝及環(huán)境配置的完整過程》,本文關(guān)鍵詞 pytorch,安裝,及,環(huán)境,配置,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。