配置
打開pycharm,文件>>設置>>項目.python解釋器,選擇pytorch的解釋器
但是,第一次使用pytorch的時候是沒有的這一項的,需要先添加上:
點擊 ⚙設置>>添加,選擇conda環境>>現有環境>>…選擇python解釋器,選擇的解釋器是pytorch環境下的python.exe解釋器
然后點擊應用、確定,靜靜等待載入配置就可以了
驗證
運行下面的代碼,看能否輸出版本號、True
import torch
print(torch.__version__)
print(torch.cuda.is_available())