一、首先安裝CUDA? ?
pytorch2.0系列目前只支持CUDA11.7與CUDA11.8，首先通過該網(wǎng)址https://developer.nvidia.com/cuda-toolkit-archive，下載對應(yīng)的CUDA11.7或者CUDA11.8，
然后雙擊安裝，安裝時選擇自定義選項
如果自己的電腦上沒有下載VSstudio,在自定義選項里面取消勾選VS選項
如果下載的組件版本低于或者等于當(dāng)前版本，也可以取消勾選。
?然后是選擇安裝位置，這里可以自己自定義安裝路徑，自己創(chuàng)建一個CUDA11.8的文件夾，然后在文件加下創(chuàng)建CUDA01與CUDA02，然后將CUDA安裝到對應(yīng)的路徑。
?二、第二步是下載cudnn
https://developer.nvidia.com/rdp/cudnn-archive
第一次下載需要注冊一個賬號，注冊即可。然后下載對應(yīng)版本的cudnn，然后解壓后會有三個文件夾
?
將三個文件夾復(fù)制到對應(yīng)的CUDA文件夾下就可以了
三、創(chuàng)建虛擬環(huán)境然后裝對應(yīng)的pytorch版本
?conda create -n 環(huán)境名?python==指定的python版本，創(chuàng)建完環(huán)境用activate 環(huán)境名 來激活環(huán)境，
然后就可以安裝包了
官方給的安裝命令
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
如果安裝太慢，也可以使用離線輪子進行安裝。
四、驗證是否安裝成功
在命令行輸入一下命令
如果顯示True則證明安裝成功。文章來源：http://www.zghlxwxcb.cn/news/detail-442019.html
聽說pytorch2.0版本可以通過torch.compile()?增加訓(xùn)練速度，然后興致勃勃的安裝了2.0版本，結(jié)果最后發(fā)現(xiàn)Windows現(xiàn)在還不支持使用compile功能，期待Windows早日支持吧。文章來源地址http://www.zghlxwxcb.cn/news/detail-442019.html
到了這里，關(guān)于pytorch2.0安裝教程的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
