cuda安裝精簡

使用torch的一定注意cuda版本必須要跟torch版本對應,網上好多網友寫的博客都沒提到,目前cuda官網的版本是11.0但是我當時在torch官網下載的是支持cuda10.2版本的torch。 cuda 10.2 鏈接是cuda10.2版本的下載鏈接,下載之後按照提示步驟一步步安裝即可非常簡單。 兩種方法測試是否安裝成功: 1.進入cmd,輸入nvcc -V,會出現cuda的版本號等相關信息
相關文章
相關標籤/搜索