虛擬環(huán)境的創(chuàng)建

命令行窗口中使用
conda create -n 環(huán)境名 python=所需python版本
即可創(chuàng)建虛擬環(huán)境

pytorch的gpu版本安裝

首先確定自己電腦的gpu版本
打開顯卡控制面板

點擊系統(tǒng)信息，選擇組件

產(chǎn)品名稱中CUDA后的11.0便是gpu版本

再在pytorch官網(wǎng)選擇對應版本

進入為pytorch創(chuàng)建的虛擬環(huán)境，輸入命令，等待片刻，pytorch便安裝完成了。
再在NVIDIA官網(wǎng)安裝cuda和cudnn

將cudnn的三個文件分別放入cuda安裝目錄下，即可完成配置
在命令行中輸入驗證代碼

如上圖所示即為成功

到此這篇關于pytorch安裝及環(huán)境配置的完整過程的文章就介紹到這了,更多相關pytorch安裝及環(huán)境配置內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！