conda pytorch 配置


主要步驟:

0.安裝anaconda3(基本沒問題)

1.配置清華的源(基本沒問題)

2.查看python版本,運行 python3 -V; 查看CUDA版本,運行 nvcc -V

3.如果想用最新版本的python,可以創建新的python版本:

  conda create --name python38 python=3.8

  conda activate python38

此時,運行python3 -V,會顯示python的版本為3.8。再運行

  conda deactivate

python的版本會恢復為原來的,如3.7

4.根據官網(https://pytorch.org/),獲取命令行。此時要注意,如果要用清華的源,需要命令行中的“-c pytorch”;然后將python版本添加到命令行中,如“python=3.8”

在我本地,運行的命令是“conda install pytorch torchvision torchaudio cudatoolkit=10.2 python=3.8”

 

 

 在我這安裝時,一開始是使用python3.7,但安裝完之后,運行 import torch 會有 ModuleNotFoundError: No module named 'typing_extensions' 的錯誤,然后發現安裝pytorch前后,python的版本由3.7.0變為3.7.9了。最后試着創建python 3.8,安裝pytorch時指定python版本為3.8,嘗試import torch后沒問題。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM