虛擬環境的創建

命令行窗口中使用
conda create -n 環境名 python=所需python版本
即可創建虛擬環境

pytorch的gpu版本安裝

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

點擊系統信息，選擇組件

產品名稱中CUDA后的11.0便是gpu版本

再在pytorch官網選擇對應版本

進入為pytorch創建的虛擬環境，輸入命令，等待片刻，pytorch便安裝完成了。
再在NVIDIA官網安裝cuda和cudnn

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

如上圖所示即為成功

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