之前安裝過GPU版本的pytorch，但是不知道什么原因虛擬環境無法使用，重新安裝是注意要查看GUDA是否存在

官方驗證步驟：https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#compiling-examples__valid-results-from-sample-cuda-bandwidthtest-program
??在此之前，可以添加環境變量。這里我按照官網上的步驟并沒有添加
First：通過nvcc -V驗證
??1、按Windows+R，輸入cmd，進去cmd.exe
??2、輸入nvcc -V
??3、顯示類型下圖的版本信息
在這里插入圖片描述
??還有另外的方法進行驗證
Second： 運行deviceQuery或bandwidthTest 示例程序

然后重新下載相應版本的cuDNN
最后重建 虛擬環境 安裝pytorch

總結

以上是生活随笔為你收集整理的重新安装Pytorch GPU版的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。